Differentiable Simulations Python Packages

differt

Differentiable Ray Tracing Toolbox for Radio Propagation Simulations

cheetah-accelerator

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.

phiflow

A differentiable PDE solving framework for machine learning

softtorch

A library for soft differentiable relaxations of common PyTorch functions.

pyrddlgym-jax

JAX compilation of RDDL description files, and a differentiable planner in JAX.

tensormesh-fem

A fast, differentiable, JIT-free, debugging-friendly finite element library for PyTorch.

agent-torch

large population models

jwave

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

nudge-bio

NUDGE (Node/edge Ultrasensitivity Diagnostic for Gene-regulatory Effects). My "Built with Claude: Life Sciences" Hackathon project.

reactix

1-D reactive transport models with JAX

jaxsso

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

socialforce

Differentiable Social Force simulation with universal interaction potentials.