3outeille / Yaae
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
- A Numpy implementation of a Generative Adversarial Network.☆17Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Image augmentation library for Jax☆39Updated last year
- Keras-like APIs for JAX framework☆50Updated 2 years ago
- 🦠 AD in less than 20 lines☆54Updated 3 years ago
- Reinforcement learning modular with pytorch☆11Updated 4 years ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- [WIP] Note on the research papers I read☆7Updated 5 years ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 3 years ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆26Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Clean up the public namespace of your package!☆55Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…☆24Updated last year
- A simple wrapper over `pydot` and `graphviz` which fixes some sharp edges☆63Updated 2 years ago
- Annotate python source code☆69Updated 5 years ago
- Create Numpy .npy files by appending on the growth axis☆63Updated 6 months ago
- ☆24Updated last year
- ☆18Updated 3 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆128Updated last year
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆39Updated 5 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- JMP is a Mixed Precision library for JAX.☆194Updated 2 months ago
- Lightweight ML Experiment Logging 📖☆80Updated 7 months ago
- Some small scale experiments for my blog posts 📝☆79Updated 2 years ago
- Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.☆60Updated 3 years ago