3outeille / YaaeLinks
Yaae: Yet another autodiff engine (written in Numpy).
☆28Updated 2 years ago
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 5 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- ☆69Updated last year
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- Documentation:☆122Updated 2 years ago
- ☆48Updated 7 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Image augmentation library for Jax☆40Updated last year
- Functional machine learning for fun☆85Updated 4 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Source code, datasets and trained models for the paper Learning Advanced Mathematical Computations from Examples (ICLR 2021), by François…☆181Updated 4 years ago
- Annotating tensor shapes using Python types☆158Updated 2 years ago
- JMP is a Mixed Precision library for JAX.☆208Updated 7 months ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆183Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆50Updated last month
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆128Updated last year
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 3 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 2 years ago
- gpwho? GPU usage logger, with Telegram bot integration.☆14Updated 5 years ago
- ☆40Updated 2 years ago
- A simple wrapper over `pydot` and `graphviz` which fixes some sharp edges☆63Updated 2 years ago
- Alphazero on GPU thanks to CUDA.jl☆32Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆52Updated 4 years ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆26Updated 3 years ago
- Python Research Framework☆106Updated 2 years ago
- Drop-in autodiff for NumPy.☆213Updated 2 weeks ago