Efficiently Composable Data Augmentation on the GPU with Jax
☆42May 16, 2025Updated 11 months ago
Alternatives and similar repositories for augmax
Users that are interested in augmax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Image augmentation library for Jax☆42Apr 9, 2024Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆436Mar 6, 2025Updated last year
- A metrics library for the JAX ecosystem☆41Mar 16, 2023Updated 3 years ago
- Library for reading and processing ML training data.☆722Updated this week
- codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"☆10Dec 30, 2024Updated last year
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- CLU lets you write beautiful training loops in JAX.☆368Mar 3, 2026Updated 2 months ago
- Unity package for Reachy 2021☆15Aug 2, 2023Updated 2 years ago
- PyTorch Lightning based framework to run experiments for self-supervised learning tasks.☆10Feb 14, 2020Updated 6 years ago
- ☆11Feb 20, 2020Updated 6 years ago
- Unofficial Implementation of Selective Attention Transformer☆20Oct 31, 2024Updated last year
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆35Apr 19, 2025Updated last year
- Tensorflow implementation of "MC2-Net: Motion Correction Network for Multi-Contrast Brain MRI"☆12Dec 8, 2022Updated 3 years ago
- Workshop that will take you from Graph Neural Networks (GNNs) to Transformers, architectures which have led to numerous breakthrough achi…☆12Sep 11, 2023Updated 2 years ago
- [WIP] Code for LangToMo☆21Mar 19, 2026Updated last month
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Active Learning of Abstract Plan Feasibility☆12Feb 10, 2023Updated 3 years ago
- ICLR 2023: Learning to Extrapolate: A Transductive Approach☆11Aug 15, 2023Updated 2 years ago
- PErception and Robotic Learning System v2☆12Mar 31, 2023Updated 3 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…☆25Aug 16, 2023Updated 2 years ago
- Parallel hyperparameter tuning with JAX☆39Apr 16, 2026Updated 2 weeks ago
- Code for the paper Normalizing Flows are Capable Models for RL☆19Jun 3, 2025Updated 11 months ago
- Custom PD controller and robot emulation for Franka-Emika Panda arms☆10Aug 25, 2020Updated 5 years ago
- Equivariant layers for RC-complement symmetry in DNA sequence data☆12Feb 24, 2022Updated 4 years ago
- ☆32Mar 4, 2021Updated 5 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆10Aug 26, 2022Updated 3 years ago
- A collection of utilities for machine learning experiments.☆11Jan 8, 2026Updated 3 months ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆14Nov 26, 2019Updated 6 years ago
- A toolbox for MRI gradient design☆16Feb 22, 2024Updated 2 years ago
- ☆634Apr 18, 2026Updated 2 weeks ago
- A framework for steering MoE models by detecting and controlling behavior-linked experts.☆33Sep 12, 2025Updated 7 months ago
- Benchopt benchmark for ResNet fitting on a classification task☆12Sep 19, 2023Updated 2 years ago
- A framework for evaluating LLMs in Atari games☆15Apr 21, 2025Updated last year
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆28Jan 14, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years ago
- ☆16Apr 26, 2023Updated 3 years ago
- Code for Semi-crowdsourced Clustering with Deep Generative Models☆12Dec 9, 2022Updated 3 years ago
- Code accompanying VarGrad: A Low-Variance Gradient Estimator for Variational Inference☆12Oct 12, 2020Updated 5 years ago
- A complete framework for training Large Language Models from scratch☆21Jan 8, 2026Updated 3 months ago
- A Pytree Module system for Deep Learning in JAX☆212Feb 26, 2023Updated 3 years ago
- [ 👾 ] ➡️ 💾 ➡️ { 🎮🕹️ } Extra Stable-Baselines3 buffer classes. Reducing RL memory usage drastically with minimal overhead.☆22Dec 9, 2025Updated 4 months ago