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
121 stars
7 watching
7 forks
Language: Python
last commit: 2 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 | 143 |
 patrick-kidger/diffrax | Provides numerical differential equation solvers using autodifferentiable and GPU-capable JAX. | 1,442 |
 differentiableuniverseinitiative/jax_cosmo | A Python library for differentiable cosmology using automatic differentiation on GPUs | 178 |
 jasperfx/lamar | An inversion of control framework designed to simplify the process of dependency injection in .NET applications | 573 |
 ucb-bar/midas | Automated framework for converting digital circuit designs into FPGA-accelerated simulators | 97 |
 sofa-framework/sofa | An open-source framework for interactive physics simulation with an emphasis on medical simulation and research. | 931 |
| ucl-bug/k-wave | A MATLAB toolbox for simulating the propagation of acoustic waves in various media | 54 |
 jax-md/jax-md | A library for simulating complex material behavior using automatic differentiation and hardware acceleration. | 1,185 |
 cesaraustralia/dynamicgrids.jl | A framework for building high-performance grid-based spatial simulations | 225 |
 nardo/tnl2 | A high-level networking API for real-time simulations with remote procedure call and object state replication primitives | 25 |
 luchris429/purejaxrl | A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality | 722 |
 tatsu-lab/alpaca_farm | A framework for simulating and evaluating reinforcement learning from human feedback methods | 782 |
 google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 152 |
 ugr-sail/sinergym | An environment framework for reinforcement learning simulation and control using EnergyPlus | 134 |
 djformby/grfics | A framework for simulating industrial control systems to educate and practice security skills through realistic 3D visualization. | 213 |