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:
- Unofficial JAX implementations of deep learning research papers☆162Jun 25, 2022Updated 4 years ago
- A metrics library for the JAX ecosystem☆41Mar 16, 2023Updated 3 years ago
- codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"☆11Dec 30, 2024Updated last year
- CLU lets you write beautiful training loops in JAX.☆369Jul 7, 2026Updated 3 weeks ago
- ISMRM Raw Data Viewer☆10Feb 12, 2020Updated 6 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- PyTorch Lightning based framework to run experiments for self-supervised learning tasks.☆10Feb 14, 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.☆756Updated this week
- ☆11Feb 20, 2020Updated 6 years ago
- ☆18Sep 3, 2024Updated last year
- Unofficial Implementation of Selective Attention Transformer☆20Oct 31, 2024Updated last year
- Tensorflow implementation of "MC2-Net: Motion Correction Network for Multi-Contrast Brain MRI"☆12Dec 8, 2022Updated 3 years ago
- ☆17Dec 14, 2024Updated last year
- [WIP] Code for LangToMo☆21Mar 19, 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.
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 4 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
- 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 2 years ago
- Code for Augment & Reduce, a scalable stochastic algorithm for large categorical distributions☆10May 16, 2018Updated 8 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
- Translate Latex Formulas to R Code☆14Sep 19, 2021Updated 4 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.
- Jupyter notebooks from our weekly (or so) hackathons☆11Dec 3, 2024Updated last year
- Implementation of N-Grammer in Flax☆17Nov 3, 2022Updated 3 years ago
- A toolbox for MRI gradient design☆17Feb 22, 2024Updated 2 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆47Jan 11, 2025Updated 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.☆59Apr 25, 2026Updated 3 months ago
- 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 6 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Example of Network Accelerated Motion Estimation and Reduction (NAMER) for motion correction in MRI☆19Apr 16, 2023Updated 3 years ago
- Implementation of Variational Networks for image reconstruction in tensorflow☆14May 11, 2020Updated 6 years ago
- many powerful tools for studying irreducible representations of SU(n), including making animations of hadron flavor-state multiplets☆13Jun 17, 2026Updated last month
- ☆644Updated this week
- 📈Implementing the ADAM optimizer from the ground up with PyTorch and comparing its performance on six 3-D objective functions (each prog…☆23Jul 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 2 years ago