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

agent-torch

large population models

jwave

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

pyrddlgym-jax

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

socialforce

Differentiable Social Force simulation with universal interaction potentials.

jaxsso

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

softtorch

A library for soft differentiable relaxations of common PyTorch functions.