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
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Neural Networks for JAX☆83Updated 5 months ago
- Compositional Linear Algebra☆465Updated last week
- 🧱 Modula software package☆173Updated last week
- A functional training loops library for JAX☆86Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆169Updated 3 months ago
- ☆214Updated 8 months ago
- A Jax-based library for designing and training transformer models from scratch.☆282Updated 6 months ago
- ☆245Updated 5 months ago
- Unofficial JAX implementations of deep learning research papers☆153Updated 2 years ago
- Running Jax in PyTorch Lightning☆89Updated 3 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆370Updated this week
- LoRA for arbitrary JAX models and functions☆135Updated last year
- Run PyTorch in JAX. 🤝☆231Updated last month
- ☆165Updated last year
- ☆112Updated last month
- Differentiable Algorithms and Algorithmic Supervision.☆113Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 8 months ago
- A simple library for scaling up JAX programs☆134Updated 4 months ago
- Graph neural networks in JAX.☆67Updated 9 months ago
- AutoBound automatically computes upper and lower bounds on functions.☆359Updated 8 months ago
- Uncertainty quantification with PyTorch☆346Updated last week
- Implementation of Flash Attention in Jax☆206Updated last year
- For optimization algorithm research and development.☆498Updated this week
- ☆60Updated 3 years ago
- git extension for {collaborative, communal, continual} model development☆208Updated 4 months ago
- ☆169Updated 3 months ago
- JMP is a Mixed Precision library for JAX.☆193Updated last month
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆177Updated last month