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
- PIX is an image processing library in JAX, for JAX.☆449Sep 29, 2026Updated last week
- Unofficial JAX implementations of deep learning research papers☆162Jun 25, 2022Updated 4 years ago
- CLU lets you write beautiful training loops in JAX.☆370Updated this week
- 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.
- ISMRM Raw Data Viewer☆10Feb 12, 2020Updated 6 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.☆787Updated this week
- ☆11Feb 20, 2020Updated 6 years ago
- ☆18Sep 3, 2024Updated 2 years ago
- The Lottery Ticket Hypothesis for Improving Pretrained Robot Diffusion and Flow Policies☆21Sep 28, 2026Updated last week
- ☆18Dec 14, 2024Updated last year
- [WIP] Code for LangToMo☆21Mar 19, 2026Updated 6 months ago
- Hückel model + JAX☆14Oct 13, 2022Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Objective metrics for measuring visual texture similarity using STSIM features. Supervised by Thrasos Pappas.☆14Oct 4, 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
- Active Learning of Abstract Plan Feasibility☆12Feb 10, 2023Updated 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
- Here is an implementation of DeepLabv3+ in PyTorch(1.7). It supports many backbones and datasets.☆17Nov 14, 2022Updated 3 years ago
- Custom PD controller and robot emulation for Franka-Emika Panda arms☆10Aug 25, 2020Updated 6 years ago
- Implementation of transformations.py that is JAX compatible☆13Mar 10, 2023Updated 3 years ago
- Equivariant layers for RC-complement symmetry in DNA sequence data☆13Feb 24, 2022Updated 4 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.
- Translate Latex Formulas to R Code☆14Sep 19, 2021Updated 5 years ago
- ☆10Aug 26, 2022Updated 4 years ago
- A toolbox for MRI gradient design☆17Feb 22, 2024Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆106Oct 23, 2023Updated 2 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆14Nov 26, 2019Updated 6 years ago
- ☆11Jan 8, 2025Updated last year
- A Unified Framework for Transforming between Text, Point Cloud, and Program☆19Jul 3, 2025Updated last year
- MUX-PLMs: Pretraining LMs with Data Multiplexing☆15Jan 29, 2023Updated 3 years ago
- Example of Network Accelerated Motion Estimation and Reduction (NAMER) for motion correction in MRI☆19Apr 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.
- Implementation of Variational Networks for image reconstruction in tensorflow☆14May 11, 2020Updated 6 years ago
- ☆19Jul 25, 2025Updated last year
- ☆657Updated this week
- 📈Implementing the ADAM optimizer from the ground up with PyTorch and comparing its performance on six 3-D objective functions (each prog…☆24Jul 2, 2022Updated 4 years 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
- Benchopt benchmark for ResNet fitting on a classification task☆12Sep 19, 2023Updated 3 years ago
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years ago