Microbenchmarking hyperparameter tuning for JAX functions.
☆21Jul 7, 2026Updated last week
Alternatives and similar repositories for tune-jax
Users that are interested in tune-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Tokamax: A GPU and TPU kernel library.☆249Updated this week
- Minimal yet performant LLM examples in pure JAX☆267Jul 4, 2026Updated 2 weeks ago
- JAX support for tvm-ffi abi☆26May 14, 2026Updated 2 months ago
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated 4 months ago
- torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…☆231Jul 3, 2026Updated 2 weeks ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆46Updated this week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆387Updated this week
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated 2 years ago
- ☆19Aug 26, 2025Updated 10 months ago
- A Python DSL for Apple Metal GPU compute☆25Jun 20, 2026Updated last month
- ☆34Jun 3, 2024Updated 2 years ago
- ☆22May 7, 2025Updated last year
- A simple library for scaling up JAX programs☆148Nov 4, 2025Updated 8 months ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆15May 10, 2024Updated 2 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.
- LES extension package for NequIP☆18May 26, 2026Updated last month
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated 2 years ago
- Virtual Decoupled Cores: Composable Programming Framework and Runtime for Async GPUs☆19Updated this week
- JMeter plugin to Publish and Subscribe Messages to GCP☆12Jan 5, 2025Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆47Jan 11, 2025Updated last year
- SpecTrax is a JAX-native library for neural networks and graph learning, built for performance, composability and modularity.☆42Jul 13, 2026Updated last week
- ☆17Apr 7, 2025Updated last year
- Homework 3 for Berkeley CS 280: our version of the MIT Mini Places challenge☆12Mar 5, 2016Updated 10 years ago
- JAX backend for SGL☆311Updated this week
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Learn one, get them all for free☆12Jan 28, 2024Updated 2 years ago
- ☆304Updated this week
- (EasyDel Former) is a utility library designed to simplify and enhance the development in JAX☆33Jun 21, 2026Updated 3 weeks ago
- A simple, performant, and scalable Jax LLM!☆2,365Updated this week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆80Jun 18, 2026Updated last month
- ☆15Jan 26, 2021Updated 5 years ago
- JAX Synergistic Memory Inspector☆186Jul 16, 2024Updated 2 years ago
- ☆16Mar 13, 2025Updated last year
- Automatic differentiation for Triton Kernels☆29Aug 12, 2025Updated 11 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- a Jax quantization library☆125Updated this week
- Stellenbosch University ZeroSpeech 2019 System☆10Apr 4, 2019Updated 7 years ago
- Machine Learning eXperiment Utilities☆48Jul 29, 2025Updated 11 months ago
- ☆18Apr 17, 2026Updated 3 months ago
- Integrates Imbue's Cost Aware pareto-Region Bayesian Search (CARBS) with Weights and Biases (WanDB)☆12Mar 17, 2025Updated last year
- Minimal, lightweight JAX implementations of popular models.☆239May 29, 2026Updated last month
- Code for implementing central flows☆48Sep 5, 2025Updated 10 months ago