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
Elegant and Performant Deep Learning
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
A PyTorch library for implementing flow matching algorithms
Qwen2.5-Coder is the code version of Qwen2.5, the large language model
Gorgonia is a library that helps facilitate machine learning in Go
simple algorithm for a realtime interactive visual cortex for painting