3outeille / Yaae
Yaae: Yet another autodiff engine (written in Numpy).
☆28Updated last year
Related projects: ⓘ
- A Numpy implementation of a Generative Adversarial Network.☆15Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆41Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆59Updated last year
- A Pytree Module system for Deep Learning in JAX☆215Updated last year
- [WIP] Note on the research papers I read☆7Updated 5 years ago
- Tidy up your machine learning experiments☆17Updated 5 years ago
- Lightweight ML Experiment Logging 📖☆79Updated 3 weeks 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 last year
- a simple implementation of autograd engine☆24Updated 5 years ago
- ⛰️ RockyML - A High-Performance Scientific Computing Framework for Non-smooth Machine Learning Problems☆16Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆100Updated 2 months ago
- ☆30Updated this week
- Image augmentation library for Jax☆36Updated 5 months ago
- A lightweight deep learning framework☆32Updated 3 years ago
- Documentation:☆119Updated last year
- A miniscule implementation of reverse mode auto-differentiation☆27Updated 3 years ago
- Running Jax in PyTorch Lightning☆76Updated last year
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…☆23Updated last year
- ☆24Updated last year
- 👑 Pytorch code for the Nero optimiser.☆20Updated last year
- Keras-like APIs for JAX framework☆50Updated last year
- ☆64Updated 10 months ago
- ☆56Updated 2 years ago
- A functional training loops library for JAX☆84Updated 7 months ago
- High performance implementation of Deep neuroevolution in pytorch using mpi4py. Intended for use on HPC clusters☆24Updated 2 years ago
- ☆18Updated 2 years ago
- 🦠 AD in less than 20 lines☆48Updated 3 years ago
- ☆38Updated last year
- Draw, modify, and execute graphs using a simple python dict interface☆29Updated last year