3outeille / YaaeLinks
Yaae: Yet another autodiff engine (written in Numpy).
☆27Updated last year
Alternatives and similar repositories for Yaae
Users that are interested in Yaae are comparing it to the libraries listed below
Sorting:
- A Numpy implementation of a Generative Adversarial Network.☆17Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Inference code in Pytorch for GPT-like models, such as PAGnol, a family of models with up to 1.5B parameters, trained on datasets in Fren…☆20Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Image augmentation library for Jax☆39Updated last year
- 🦠 AD in less than 20 lines☆54Updated 3 years ago
- Documentation:☆120Updated 2 years ago
- Some small scale experiments for my blog posts 📝☆79Updated 2 years ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- Keras-like APIs for JAX framework☆49Updated 2 years ago
- Reinforcement learning modular with pytorch☆11Updated 4 years ago
- ⛰️ RockyML - A High-Performance Scientific Computing Framework for Non-smooth Machine Learning Problems☆19Updated 2 years ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Updated 5 years ago
- ☆18Updated 3 years ago
- PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆37Updated 3 years ago
- Symbolic API for model creation in PyTorch.☆66Updated 2 months ago
- A JAX implementation of stochastic addition.☆14Updated 2 years ago
- Functional machine learning for fun☆85Updated 4 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆59Updated 5 years ago
- ☆108Updated 2 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- A minimalist neural networks library built on a tiny autograd engine☆19Updated 11 months ago
- High performance implementation of Deep neuroevolution in pytorch using mpi4py. Intended for use on HPC clusters☆27Updated 3 years ago
- A stateful pytree library for training neural networks.☆22Updated 2 years ago
- Lightweight ML Experiment Logging 📖☆80Updated 9 months ago