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
- 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
- CLU lets you write beautiful training loops in JAX.☆371Jul 7, 2026Updated 2 months ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Unity package for Reachy 2021☆15Aug 2, 2023Updated 3 years ago
- This is the repository for the paper Static Prediction of Runtime Errors by Learning to Execute Programs with External Resource Descripti…☆25Nov 18, 2022Updated 3 years ago
- Library for reading and processing ML training data.☆781Updated this week
- Unofficial Implementation of Selective Attention Transformer☆20Oct 31, 2024Updated last year
- ☆17Dec 14, 2024Updated last year
- [WIP] Code for LangToMo☆21Mar 19, 2026Updated 5 months ago
- Hückel model + JAX☆14Oct 13, 2022Updated 3 years ago
- PErception and Robotic Learning System v2☆12Mar 31, 2023Updated 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 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.
- Code for Augment & Reduce, a scalable stochastic algorithm for large categorical distributions☆10May 16, 2018Updated 8 years ago
- Custom PD controller and robot emulation for Franka-Emika Panda arms☆10Aug 25, 2020Updated 6 years ago
- Integrates Imbue's Cost Aware pareto-Region Bayesian Search (CARBS) with Weights and Biases (WanDB)☆12Mar 17, 2025Updated last year
- Equivariant layers for RC-complement symmetry in DNA sequence data☆13Feb 24, 2022Updated 4 years ago
- Translate Latex Formulas to R Code☆14Sep 19, 2021Updated 4 years ago
- Jupyter notebooks from our weekly (or so) hackathons☆11Dec 3, 2024Updated last year
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆14Nov 26, 2019Updated 6 years ago
- ☆32Mar 4, 2021Updated 5 years ago
- [ICML2025] Official implementation of Efficient Online Reinforcement Learning for Diffusion Policies appearing in ICML 2025.☆61Apr 25, 2026Updated 4 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.
- ☆11Jan 8, 2025Updated last year
- A Unified Framework for Transforming between Text, Point Cloud, and Program☆19Jul 3, 2025Updated last year
- A collection of utilities for machine learning experiments.☆11Jan 8, 2026Updated 8 months ago
- MUX-PLMs: Pretraining LMs with Data Multiplexing☆15Jan 29, 2023Updated 3 years ago
- Sample and predict coarse grain 3D RNA structures☆10Feb 12, 2024Updated 2 years ago
- many powerful tools for studying irreducible representations of SU(n), including making animations of hadron flavor-state multiplets☆13Jun 17, 2026Updated 3 months ago
- ☆654Updated this week
- Benchopt benchmark for ResNet fitting on a classification task☆12Sep 19, 2023Updated 2 years ago
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years 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.
- A framework for evaluating LLMs in Atari games☆15Apr 21, 2025Updated last year
- Procedural object generation for robotic manipulation☆11Oct 6, 2018Updated 7 years ago
- JMP is a Mixed Precision library for JAX.☆214Updated this week
- JAX interpreter for Vulkan☆16Jun 1, 2021Updated 5 years ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Jun 14, 2021Updated 5 years ago
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆35Apr 19, 2025Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆113Jul 19, 2024Updated 2 years ago