Differentiable Physics Python Packages

exponax

Efficient Differentiable n-d PDE Solvers in JAX.

apebench

[Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled Training; Rollout Metrics)

tesseract-core

Universal components for differentiable scientific computing. :package:

neuralgcm-torch

PyTorch port of NeuralGCM

eiko

Eiko: the GPU-accelerated Eikonal equation solver

dmg

Generic framework for building differentiable models.

dinosaur-torch

PyTorch port of NeuralGCM

softtorch

A library for soft differentiable relaxations of common PyTorch functions.

synth-afm

Differentiable High-Speed AFM (HS-AFM) simulation for protein structures. JAX-powered toolkit for generating synthetic movies with realistic tip-dilation and scanning lag.

mosaic-bench

A benchmark suite and re-usable collection of differentiable physics solvers.

hydrodl2

Differentiable Hydrologic Model Library

jaxion

A simple JAX-powered simulation library for numerical experiments of fuzzy dark matter, stars, gas + more!

adirondax

A differentiable astrophysics simulator in JAX

pycharge

Electromagnetics simulation library for moving point charges built on JAX

hydrologeez

Differentiable conceptual hydrological models in JAX/Equinox — end-to-end gradients through the full simulation for parameter calibration.

softjax

A library for soft differentiable relaxations of common JAX functions.

spicex

A differentiable SPICE circuit solver in JAX

updes

Universal Partial Differential Equations Simulator

pychargess

Electromagnetics simulation library for moving point charges built on JAX