Popular repositories Loading

1. lineax lineax Public

   Forked from patrick-kidger/lineax

   Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

   Python

2. adept adept Public

   Forked from ergodicio/adept

   Automatic-Differentiation-Enabled Plasma Transport in JAX

   Python

3. optimistix optimistix Public

   Forked from patrick-kidger/optimistix

   Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

   Python

4. diffrax diffrax Public

   Forked from patrick-kidger/diffrax

   Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

   Python

5. podman-hpc podman-hpc Public

   Forked from NERSC/podman-hpc

   Python

6. tesseract-jax tesseract-jax Public

   Forked from pasteurlabs/tesseract-jax

   Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡

   Python