jxbz / agd
Automatic gradient descent
☆206Updated last year
Related projects ⓘ
Alternatives and complementary repositories for agd
- ☆105Updated 2 weeks ago
- Run PyTorch in JAX. 🤝☆201Updated this week
- ☆197Updated 4 months ago
- Neural Networks for JAX☆83Updated last month
- A simple library for scaling up JAX programs☆127Updated 3 weeks ago
- ☆207Updated 7 months ago
- ☆240Updated last month
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- Unofficial JAX implementations of deep learning research papers☆151Updated 2 years ago
- LoRA for arbitrary JAX models and functions☆133Updated 8 months ago
- AutoBound automatically computes upper and lower bounds on functions.☆355Updated 4 months ago
- A functional training loops library for JAX☆85Updated 9 months ago
- JAX Synergistic Memory Inspector☆164Updated 4 months ago
- git extension for {collaborative, communal, continual} model development☆205Updated last week
- Train very large language models in Jax.☆195Updated last year
- JAX Arrays for human consumption☆88Updated last year
- Scalable neural net training via automatic normalization in the modular norm.☆122Updated this week
- A Jax-based library for designing and training transformer models from scratch.☆276Updated 2 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆95Updated 6 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆128Updated last month
- A Python package of computer vision models for the Equinox ecosystem.☆102Updated 4 months ago
- JMP is a Mixed Precision library for JAX.☆187Updated 6 months ago
- ☆152Updated 11 months ago
- An interactive exploration of Transformer programming.☆247Updated last year
- Compositional Linear Algebra☆432Updated this week
- ☆129Updated last week
- ☆58Updated 2 years ago
- ☆293Updated 5 months ago
- Running Jax in PyTorch Lightning☆82Updated 2 weeks ago