Popular repositories Loading

1. neojax neojax Public

   Neural Operators in Jax

   Python 4 2

2. jax-fem jax-fem Public

   Forked from deepmodeling/jax-fem

   Differentiable Finite Element Method with JAX

   Python

3. equinox equinox Public

   Forked from patrick-kidger/equinox

   Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

   Python

4. awesome-jax awesome-jax Public

   Forked from n2cholas/awesome-jax

   JAX - A curated list of resources https://github.com/google/jax