7 projects
chaotax
Simple JAX-based tool to compute Lyapunov spectra and exponents of dynamical systems.
picardax
Time-Implicit Differentiable Finite Difference Solvers in JAX.
exponax
Efficient differentiable PDE solvers in JAX.
vape4d
Volume 4D Visualization
apebench
Benchmark suite for Autoregressive Neural Emulators of PDEs in JAX.
trainax
Training methodologies for Autoregressive Neural Emulators in JAX built on top of Equinox & Optax.
pdequinox
Neural PDE Emulator Architectures in JAX built on top of Equinox.
