Pinned Loading

1. kernex kernex Public

   Stencil computations in JAX

   Python 66 3

2. DiffusionNet DiffusionNet Public

   Solving PDEs using Deep learning Code Associated with https://arxiv.org/abs/2011.10015

   Jupyter Notebook 7 1

3. serket serket Public

   The ✨Magical✨ JAX ML Library.

   Python 17

4. Physics-informed-neural-network-in-JAX Physics-informed-neural-network-in-JAX Public

   Example problems in Physics informed neural network in JAX

   Jupyter Notebook 72 6

5. finitediffX finitediffX Public

   Finite difference tools in JAX

   Python 8 3