Stars
Geometric kernels on manifolds, meshes and graphs
Massively parallel training of machine-learning based weather and climate models
Deep-learning framework for Engineering AI. Built on transformer building blocks, it delivers the full engineering stack, allowing teams to build, train, and operate industrial simulation models ac…
Plug-and-Play Benchmarking of Reinforcement Learning Algorithms for Large-Scale Flow Control
1D-1V Vlasov-Poisson(-Fokker-Planck), Plasma Physics PDE Simulation Tool in NumPy and experiment management in MLFlow
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