Pre-built implicit layer architectures with O(1) backprop, GPUs
A package for the sparse identification of nonlinear dynamical systems
Numerical differential equation solvers in JAX
Data driven modeling and automated discovery of dynamical systems
Open-source deep-learning framework for building and training
A library for scientific machine learning & physics-informed learning
Physics-Informed Neural Networks (PINN) Solvers
Optax is a gradient processing and optimization library for JAX
Gorgonia is a library that helps facilitate machine learning in Go