Efficiently Composable Data Augmentation on the GPU with Jax
☆42May 16, 2025Updated 9 months ago
Alternatives and similar repositories for augmax
Users that are interested in augmax are comparing it to the libraries listed below
Sorting:
- Image augmentation library for Jax☆41Apr 9, 2024Updated last year
- codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"☆10Dec 30, 2024Updated last year
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated last year
- Unity package for Reachy 2021☆15Aug 2, 2023Updated 2 years ago
- JAX implementation of Kolmogorov Arnold Networks (KANs).☆10May 7, 2024Updated last year
- Unofficial JAX implementations of deep learning research papers☆161Jun 25, 2022Updated 3 years ago
- Transformer creation from scratch using Jax.☆19Dec 8, 2024Updated last year
- Library for reading and processing ML training data.☆685Feb 27, 2026Updated last week
- CLU lets you write beautiful training loops in JAX.☆367Updated this week
- Unofficial Implementation of Selective Attention Transformer☆20Oct 31, 2024Updated last year
- [WIP] Code for LangToMo☆20Jun 25, 2025Updated 8 months ago
- Parallel hyperparameter tuning with JAX☆39Jul 21, 2025Updated 7 months ago
- This repository contains code for experiments using the winning mode for xView2 "xView2: Assess Building Damage" challenge and a simple …☆26Aug 8, 2023Updated 2 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
- Codebase for fine-tuning Llama2 70B to generate math test questions and answers.☆11Aug 30, 2024Updated last year
- JMP is a Mixed Precision library for JAX.☆211Jan 30, 2025Updated last year
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- Concurrency library☆17Oct 13, 2024Updated last year
- ☆11Dec 23, 2024Updated last year
- ☆32Mar 4, 2021Updated 5 years ago
- Library implementation of "No Train, all Gain: Self-Supervised Gradients Improve Deep Frozen Representations"☆40Oct 31, 2024Updated last year
- Interpretable ML for TabPFN☆47Jul 13, 2025Updated 7 months ago
- ☆22Feb 19, 2026Updated 2 weeks ago
- CANdle - a library for using USB-FDCAN dongle and communicating with md80 drives☆15Sep 15, 2025Updated 5 months ago
- ☆10Apr 7, 2024Updated last year
- Models for packages and the resources they contain.☆14Mar 10, 2024Updated last year
- This is a python library. Install with "python3 -m pip install rp" then run with "python3 -m rp" or just "rp". Requires python≥3.5☆13Feb 16, 2026Updated 2 weeks ago
- ☆10Jan 5, 2023Updated 3 years ago
- An active inference model of Lacanian psychoanalysis☆15Jun 7, 2025Updated 8 months ago
- Original VinVL visual backbone with simplified APIs to easily extract features, boxes, object detections, in a few lines of Python code.☆11Nov 27, 2022Updated 3 years ago
- This repository contains the implementation for our work "TopoDiffusionNet: A Topology-aware Diffusion Model", accepted to ICLR 2025.☆21Apr 17, 2025Updated 10 months ago
- Material parsers and other tools, scripts Initially developed for Grobid Superconductor☆13Feb 21, 2025Updated last year
- YouTube Assistant☆12May 15, 2023Updated 2 years ago
- Develop C++/CUDA extensions with PyTorch like Python scripts☆10Jan 7, 2026Updated last month
- Minimal Implementation of Bayesian Optimization in JAX☆104Apr 24, 2025Updated 10 months ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆44Jun 14, 2021Updated 4 years ago
- Pytorch-like dataloaders for JAX.☆101Dec 16, 2025Updated 2 months ago
- ☆13Nov 27, 2025Updated 3 months ago
- Position Coupling: Improving Length Generalization of Arithmetic Transformers Using Task Structure (NeurIPS 2024) + Arithmetic Transfor…☆14Oct 26, 2025Updated 4 months ago