3outeille / YaaeLinks
Yaae: Yet another autodiff engine (written in Numpy).
☆27Updated 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 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Keras-like APIs for JAX framework☆50Updated 2 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- ☆39Updated 2 years ago
- Image augmentation library for Jax☆39Updated last year
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- JMP is a Mixed Precision library for JAX.☆208Updated 7 months 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
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- Small deep learning library written from scratch in Python, using NumPy/CuPy.☆124Updated 2 years ago
- Create real-time plots in Jupyter Notebooks.☆142Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated last month
- ☆69Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆108Updated last year
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Documentation:☆121Updated 2 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 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
- Productionize machine learning predictions, with ONNX or without☆66Updated last year
- A simple wrapper over `pydot` and `graphviz` which fixes some sharp edges☆63Updated 2 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- This project was moved to: https://github.com/coax-dev/coax☆160Updated 2 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- PyTorch interface for the IPU☆180Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆178Updated 3 months ago
- Functional machine learning for fun☆85Updated 4 years ago
- Annotating tensor shapes using Python types☆158Updated 2 years ago