Stars
Code repository for our paper DiffCloth: Differentiable Cloth Simulation with Dry Frictional Contact
Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.
Efficient Differentiable n-d PDE solvers in JAX.
[Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled Training; Rollout Metrics)
Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
High performance computational platform in Python for the spectral Galerkin method
JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.
Implementation of the Surya Foundation Model and Downstream Tasks for Heliophysics
[ICLR 2024] DiffTactile: A Physics-based Differentiable Tactile Simulator for Contact-rich Robotic Manipulation
Learning in infinite dimension with neural operators.
A Large-Scale Multimodal Car Dataset with Computational Fluid Dynamics Simulations and Deep Learning Benchmarks
A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
Data-driven reduced order modeling for nonlinear dynamical systems
Data-driven reduced order modeling for nonlinear dynamical systems
Generalized sparse regression for continuous and discrete data
Apply hard physical constraints directly to Reservoir Computers!
differentiable (magneto)hydrodynamics for astrophysics in JAX
A multi-agent reinforcement learning environment to design and benchmark control strategies aimed at reducing drag in turbulent open channel flow
XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML
A generative world for general-purpose robotics & embodied AI learning.
[ICLR 2023] FluidLab: A Differentiable Environment for Benchmarking Complex Fluid Manipulation