Efficiently Composable Data Augmentation on the GPU with Jax
☆42May 16, 2025Updated 10 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:
- ☆13Dec 14, 2024Updated last year
- JAX implementation of Kolmogorov Arnold Networks (KANs).☆10May 7, 2024Updated last year
- Library for reading and processing ML training data.☆694Mar 20, 2026Updated last week
- 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
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- ☆17Sep 3, 2024Updated last year
- ☆11Feb 20, 2020Updated 6 years ago
- Unofficial Implementation of Selective Attention Transformer☆21Oct 31, 2024Updated last year
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆35Apr 19, 2025Updated 11 months ago
- Hückel model + JAX☆13Oct 13, 2022Updated 3 years ago
- [WIP] Code for LangToMo☆20Mar 19, 2026Updated last week
- 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 2 years ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆19Mar 28, 2022Updated 3 years ago
- Parallel hyperparameter tuning with JAX☆39Updated this week
- Replication of the paper "Bayesian Convolutional Neural Networks with Bernoulli Approximate Variational Inference" - Yarin Gal, Zoubin Gh…☆10Mar 23, 2018Updated 8 years ago
- Custom PD controller and robot emulation for Franka-Emika Panda arms☆10Aug 25, 2020Updated 5 years ago
- A framework for steering MoE models by detecting and controlling behavior-linked experts.☆31Sep 12, 2025Updated 6 months ago
- ECCV 2020: "Self-adapting confidence estimation for stereo"☆19Oct 27, 2020Updated 5 years ago
- Equivariant layers for RC-complement symmetry in DNA sequence data☆12Feb 24, 2022Updated 4 years ago
- ☆11Jan 8, 2025Updated last year
- ☆10Aug 26, 2022Updated 3 years ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Jupyter notebooks from our weekly (or so) hackathons☆11Dec 3, 2024Updated last year
- A collection of utilities for machine learning experiments.☆11Jan 8, 2026Updated 2 months ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆14Nov 26, 2019Updated 6 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆46Jan 11, 2025Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago
- many powerful tools for studying irreducible representations of SU(n), including making animations of hadron flavor-state multiplets☆13Jul 25, 2021Updated 4 years ago
- Procedural object generation for robotic manipulation☆11Oct 6, 2018Updated 7 years ago
- Official Implementation of SFM and the baselines in Jax.☆21May 31, 2025Updated 9 months ago
- Repository of the DynaGuide project☆51Dec 10, 2025Updated 3 months ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- JMP is a Mixed Precision library for JAX.☆212Jan 30, 2025Updated last year
- ☆627Jan 23, 2026Updated 2 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 11 months ago
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆32Oct 26, 2022Updated 3 years ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆27Jan 14, 2025Updated last year
- ☆17Sep 27, 2023Updated 2 years ago