Optimal feedback control + interventions, in JAX.
-
Updated
Jul 18, 2024 - Python
Optimal feedback control + interventions, in JAX.
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Multiple dispatch over abstract array types in JAX.
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
MLP classifier on the MNIST dataset implemented in JAX with a GUI for entering hyperparameters, and a custom visualization of runs on TensorBoard.
Markov Chain Cubature via JAX. Efficient SDE solving and Bayesian inference.
Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
Simple Python application to observe effects of obliquity and eccentricity on equation of time and analemma
Add a description, image, and links to the equinox topic page so that developers can more easily learn about it.
To associate your repository with the equinox topic, visit your repo's landing page and select "manage topics."