Differentiable Diffusion Solver in JAX

This repo provides code that solves the diffusion equation in 1D and 2D

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
diffdiffusion		diffdiffusion
1d.ipynb		1d.ipynb
2d.ipynb		2d.ipynb
LICENSE		LICENSE
README.md		README.md
requirements.txt		requirements.txt

Provide feedback