jxbz / agd
Automatic gradient descent
☆207Updated last year
Alternatives and similar repositories for agd:
Users that are interested in agd are comparing it to the libraries listed below
- ☆220Updated last month
- ☆212Updated 7 months ago
- ☆112Updated last month
- AutoBound automatically computes upper and lower bounds on functions.☆359Updated 8 months ago
- ☆245Updated 5 months ago
- Run PyTorch in JAX. 🤝☆230Updated 3 weeks ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- A Jax-based library for designing and training transformer models from scratch.☆282Updated 6 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆169Updated 2 months ago
- A functional training loops library for JAX☆86Updated last year
- LoRA for arbitrary JAX models and functions☆135Updated last year
- JMP is a Mixed Precision library for JAX.☆193Updated last month
- A simple library for scaling up JAX programs☆134Updated 4 months ago
- JAX Synergistic Memory Inspector☆168Updated 7 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆369Updated this week
- JAX implementation of the Llama 2 model☆216Updated last year
- Train very large language models in Jax.☆203Updated last year
- Neural Networks for JAX☆83Updated 5 months ago
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆165Updated last year
- ☆301Updated 8 months ago
- Compositional Linear Algebra☆464Updated last month
- Running Jax in PyTorch Lightning☆89Updated 2 months ago
- ☆165Updated last year
- JAX Arrays for human consumption☆90Updated last year
- 🧱 Modula software package☆169Updated this week
- Unofficial JAX implementations of deep learning research papers☆153Updated 2 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆187Updated 2 years ago
- Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"☆477Updated 2 years ago
- ☆154Updated last year
- σ-GPT: A New Approach to Autoregressive Models☆61Updated 6 months ago