Pinned Loading

1. bonsai bonsai Public

   Forked from jax-ml/bonsai

   Minimal, lightweight JAX implementations of popular models.

   Jupyter Notebook

2. jax-ml/bonsai jax-ml/bonsai Public

   Minimal, lightweight JAX implementations of popular models.

   Jupyter Notebook 112 16

3. flax flax Public

   Forked from google/flax

   Flax is a neural network library for JAX that is designed for flexibility.

   Jupyter Notebook