jaxley

A differentiable PDE solving framework for machine learning

large population models

Pytorch differentiable molecular dynamics

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

Efficient GPU-computing simulation for differentiable crystal plasticity finite element method

A differentiable finite element analysis solver for structural optimization based on JAX

Implementation of a Spiking Neural Network in Tensorflow.

EP-PINNs implementation for 1D and 2D forward and inverse solvers for the Aliev-Panfilov cardiac electrophysiology model. Also includes Matlab finite-differences solver for data generation.

an attempt to model reaction diffusion using particles

Sequester C as LIFE.