4rtemi5 / imaxLinks
Image augmentation library for Jax
☆39Updated last year
Alternatives and similar repositories for imax
Users that are interested in imax are comparing it to the libraries listed below
Sorting:
- A Python package of computer vision models for the Equinox ecosystem.☆107Updated 11 months ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆110Updated 3 years ago
- ☆114Updated 3 weeks ago
- ☆104Updated 11 months ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- JAX Arrays for human consumption☆93Updated last week
- A functional training loops library for JAX☆88Updated last year
- Efficiently Composable Data Augmentation on the GPU with Jax☆37Updated last month
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- ☆60Updated 3 years ago
- A JAX implementation of stochastic addition.☆14Updated 2 years ago
- Documentation:☆120Updated 2 years ago
- Pytrees + dataclasses ❤️☆63Updated last month
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- A Python library for defining flat or nested dictionary flags.☆31Updated last year
- Tools for JAX☆47Updated this week
- JMP is a Mixed Precision library for JAX.☆203Updated 4 months ago
- ☆80Updated 3 years ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆41Updated 9 months ago
- A port of muP to JAX/Haiku☆25Updated 2 years ago
- ☆157Updated last year
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- Neural Networks for JAX☆84Updated 9 months ago
- Jax SSM Library☆49Updated 2 years ago
- Normalizing Flows using JAX☆83Updated last year
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- ☆192Updated 2 months ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago