metadata
tags:
- jax
- from-scratch
- educational
license: mit
Gradling
Deep learning components and models built from scratch in JAX. Code: https://github.com/tdfirth/gradling
Structure
Each top-level directory is a model architecture. Within each, subdirectories are experiments with their own configs and checkpoints.