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☆43Apr 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.☆441Jun 2, 2026Updated last month
- Unofficial JAX implementations of deep learning research papers☆162Jun 25, 2022Updated 4 years ago
- Library for reading and processing ML training data.☆747Updated this week
- 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.
- 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.☆368Updated this week
- 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
- ☆11Feb 20, 2020Updated 6 years ago
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆35Apr 19, 2025Updated last year
- 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
- Hückel model + JAX☆14Oct 13, 2022Updated 3 years ago
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 4 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Objective metrics for measuring visual texture similarity using STSIM features. Supervised by Thrasos Pappas.☆14Oct 4, 2023Updated 2 years ago
- ☆19Mar 28, 2022Updated 4 years ago
- PErception and Robotic Learning System v2☆12Mar 31, 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 2 years ago
- Parallel hyperparameter tuning with JAX☆39May 21, 2026Updated last month
- Code for the paper Normalizing Flows are Capable Models for RL☆20Jun 3, 2025Updated last year
- Code for Augment & Reduce, a scalable stochastic algorithm for large categorical distributions☆10May 16, 2018Updated 8 years ago
- Replication of the paper "Bayesian Convolutional Neural Networks with Bernoulli Approximate Variational Inference" - Yarin Gal, Zoubin Gh…☆11Mar 23, 2018Updated 8 years ago
- Here is an implementation of DeepLabv3+ in PyTorch(1.7). It supports many backbones and datasets.☆17Nov 14, 2022Updated 3 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Custom PD controller and robot emulation for Franka-Emika Panda arms☆10Aug 25, 2020Updated 5 years ago
- ☆54Aug 27, 2024Updated last year
- 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
- ☆11Jan 8, 2025Updated last year
- ☆10Aug 26, 2022Updated 3 years ago
- Jupyter notebooks from our weekly (or so) hackathons☆11Dec 3, 2024Updated last year
- A collection of utilities for machine learning experiments.☆11Jan 8, 2026Updated 6 months ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆14Nov 26, 2019Updated 6 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆47Jan 11, 2025Updated last year
- Sample and predict coarse grain 3D RNA structures☆10Feb 12, 2024Updated 2 years ago
- Procedural object generation for robotic manipulation☆11Oct 6, 2018Updated 7 years ago
- many powerful tools for studying irreducible representations of SU(n), including making animations of hadron flavor-state multiplets☆13Jun 17, 2026Updated 3 weeks ago
- JMP is a Mixed Precision library for JAX.☆212Updated this week
- A framework for evaluating LLMs in Atari games☆15Apr 21, 2025Updated last year