Bundle Solution for PDEs

Currently, there is a bundle solution available for ODEs. 
https://colab.research.google.com/drive/1L9NGeLoDONWmCr_tZkjIJziF3yo1OJHE?usp=sharing
Is it possible to make a bundle solution for PDEs such as diffusion?

So for example the PDE c' + D*cxx = 0 would be trained to get a network c(t,x,c0) with t \in [0,1], x\in [0,2], c0 \in [0,3].

Thank you!