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
- Image augmentation library for Jax☆39Updated last year
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Tidy up your machine learning experiments☆17Updated 5 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- Keras-like APIs for JAX framework☆49Updated 2 years ago
- Reinforcement learning modular with pytorch☆11Updated 4 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 small library for creating and manipulating custom JAX Pytree classes☆56Updated 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
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆51Updated 2 months ago
- Create Numpy .npy files by appending on the growth axis☆63Updated 3 weeks ago
- Annotating tensor shapes using Python types☆159Updated last year
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- [WIP] Note on the research papers I read☆7Updated 6 years ago
- 🦠 AD in less than 20 lines☆54Updated 3 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- A stateful pytree library for training neural networks.☆22Updated 2 years ago
- High performance implementation of Deep neuroevolution in pytorch using mpi4py. Intended for use on HPC clusters☆27Updated 3 years ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆26Updated 3 years ago
- A queue service for quickly developing scripts that use all your GPUs efficiently☆85Updated 2 years ago
- ☆18Updated 3 years ago
- Documentation:☆120Updated 2 years ago
- Reinforcement learning in TensorFlow 2☆22Updated 3 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- ☆24Updated last year
- Symbolic API for model creation in PyTorch.☆66Updated 3 months ago