Efficiently Composable Data Augmentation on the GPU with Jax
☆42May 16, 2025Updated last year
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☆44Apr 9, 2024Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆452Updated this week
- Transformer creation from scratch using Jax.☆20Dec 8, 2024Updated last year
- Unofficial JAX implementations of deep learning research papers☆162Jun 25, 2022Updated 4 years ago
- A metrics library for the JAX ecosystem☆40Mar 16, 2023Updated 3 years ago
- AI Agents on 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.
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆32Oct 26, 2022Updated 4 years ago
- Reinforcement learning agents that can beat a human in a game of laser hockey. Team project on the course Reinforcement Learning WS 20/21…☆10Mar 18, 2021Updated 5 years ago
- codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"☆11Dec 30, 2024Updated last year
- Personal Python toolbox☆18Feb 2, 2026Updated 8 months ago
- Diffusion Probabilistic Model in Jax☆14Apr 20, 2024Updated 2 years ago
- Code for "What really matters in matrix-whitening optimizers?"☆25Oct 31, 2025Updated 11 months ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆29Jan 14, 2025Updated last year
- Library for reading and processing ML training data.☆799Updated this week
- ISMRM Raw Data Viewer☆10Feb 12, 2020Updated 6 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- ☆11Feb 20, 2020Updated 6 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆267Sep 4, 2026Updated last month
- Unity package for Reachy 2021☆15Aug 2, 2023Updated 3 years ago
- ☆36Sep 11, 2026Updated last month
- A Pytree Module system for Deep Learning in JAX☆210Feb 26, 2023Updated 3 years ago
- Unofficial Implementation of Selective Attention Transformer☆20Oct 31, 2024Updated last year
- ☆18Dec 14, 2024Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 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 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Reinforcement Learning Algorithms☆10Nov 5, 2017Updated 8 years ago
- A web based platform for collecting human actions in reinforcement learning environments☆31Oct 16, 2026Updated last week
- JAX for Graphcore IPU (experimental)☆21Mar 12, 2024Updated 2 years ago
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 5 years ago
- Qwen3 MoE Router Statistics | 큐웬3 MoE☆11May 16, 2025Updated last year
- Tools for aligned reconstruction of multi-shot multi-slice MR.☆12Apr 10, 2017Updated 9 years ago
- Hückel model + JAX☆14Oct 13, 2022Updated 4 years ago
- Code for Augment & Reduce, a scalable stochastic algorithm for large categorical distributions☆10May 16, 2018Updated 8 years ago
- A simple library for scaling up JAX programs☆149Nov 4, 2025Updated 11 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆29Jan 12, 2022Updated 4 years ago
- [WIP] Code for LangToMo☆21Mar 19, 2026Updated 7 months ago
- ☆10Jun 5, 2023Updated 3 years ago
- Learning for effective and efficient bilevel planning; integrated with BEHAVIOR and iGibson☆12Jan 21, 2024Updated 2 years ago
- A jax-based reimplementation of MBPO☆10May 8, 2025Updated last year
- Implementation of transformations.py that is JAX compatible☆13Mar 10, 2023Updated 3 years ago
- ☆10Aug 26, 2022Updated 4 years ago