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
- Reinforcement learning modular with pytorch☆11Updated 4 years ago
- Image augmentation library for Jax☆38Updated 11 months ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- [WIP] Note on the research papers I read☆7Updated 5 years ago
- ☆24Updated last year
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 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.☆51Updated 2 months ago
- Provides symbolic API for model creation in PyTorch.☆63Updated 6 months ago
- Tidy up your machine learning experiments☆17Updated 5 years ago
- Hash-routed Networks☆20Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Massively Parallel and Asynchronous Architecture for Logic-based AI☆41Updated 2 years ago
- A Google Chrome extension that notifies you when longer running Google Colaboraty cells are finished☆22Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Annotating tensor shapes using Python types☆158Updated last year
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆39Updated 5 years ago
- ☆21Updated 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
- ☆18Updated 2 years ago
- High performance implementation of Deep neuroevolution in pytorch using mpi4py. Intended for use on HPC clusters☆26Updated 3 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 7 months ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- Autograd Engine for Python Written in Python C-API for Speed☆20Updated last year
- A framework for implementing equivariant DL☆10Updated 3 years ago
- A Pytree Module system for Deep Learning in JAX☆213Updated 2 years ago
- A Numpy implementation of a Convolutional Neural Network: slow & fast (im2col/col2im).☆56Updated last year
- A metrics library for the JAX ecosystem☆40Updated last year
- Alphazero on GPU thanks to CUDA.jl☆32Updated 3 years ago