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
- Image augmentation library for Jax☆41Updated last year
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- Documentation:☆125Updated 2 years ago
- Some small scale experiments for my blog posts 📝☆80Updated 3 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Updated 6 years ago
- gpwho? GPU usage logger, with Telegram bot integration.☆14Updated 5 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 4 years ago
- ⛰️ RockyML - A High-Performance Scientific Computing Framework for Non-smooth Machine Learning Problems☆20Updated 2 years ago
- Functional machine learning for fun☆85Updated 4 years ago
- Keras-like APIs for JAX framework☆50Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- ☆120Updated this week
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- Drop-in autodiff for NumPy.☆215Updated last month
- ☆71Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Butterfly matrix multiplication in PyTorch☆178Updated 2 years ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆28Updated 4 years ago
- ☆40Updated 3 years ago
- Annotating tensor shapes using Python types☆159Updated 2 years ago
- A simple wrapper over `pydot` and `graphviz` which fixes some sharp edges☆63Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- a simple implementation of autograd engine☆24Updated 7 years ago
- Repository of machine learning benchmarks☆49Updated 2 months ago
- Introduction to Evolutionary Computation class☆35Updated 8 months ago
- ☆48Updated 7 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…☆25Updated 2 years ago
- Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing☆14Updated 3 years ago