jaxdf
Numerical simulator framework
A JAX-based framework for creating differentiable numerical simulators with arbitrary discretizations.
A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
124 stars
7 watching
7 forks
Language: Python
last commit: 4 months ago
Linked from 1 awesome list
differentiable-programmingdiscretizationjaxpde
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
| ucl-bug/jwave | A JAX-based framework for simulating acoustic wave propagation in differentiable and parallelizable ways | 145 |
 patrick-kidger/diffrax | Provides numerical differential equation solvers using autodifferentiable and GPU-capable JAX. | 1,480 |
 differentiableuniverseinitiative/jax_cosmo | A Python library for differentiable cosmology using automatic differentiation on GPUs | 182 |
 jasperfx/lamar | A fast inversion of control tool with a focus on simplicity and ease of use. | 577 |
 ucb-bar/midas | Automated framework for converting digital circuit designs into FPGA-accelerated simulators | 98 |
 sofa-framework/sofa | An open-source framework for real-time multi-physics simulation with an emphasis on medical and robotics applications. | 941 |
| ucl-bug/k-wave | A MATLAB toolbox for simulating the propagation of acoustic waves in various media | 55 |
 jax-md/jax-md | A library for simulating molecular dynamics with automatic hardware acceleration and end-to-end differentiability using the JAX framework | 1,204 |
 cesaraustralia/dynamicgrids.jl | A framework for building high-performance grid-based spatial simulations | 226 |
 nardo/tnl2 | A high-level networking API for real-time simulations with remote procedure call and object state replication primitives | 26 |
 luchris429/purejaxrl | A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality | 755 |
 tatsu-lab/alpaca_farm | A framework for simulating and evaluating reinforcement learning from human feedback methods | 786 |
 google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 153 |
 ugr-sail/sinergym | A Gymnasium-based interface for interacting with simulation engines like EnergyPlus to build and test control strategies using reinforcement learning | 137 |
 djformby/grfics | A framework for simulating industrial control systems to educate and practice security skills through realistic 3D visualization. | 215 |