SpecTrax is a JAX-native library for neural networks and graph learning, built for performance, composability and modularity.
☆43Aug 15, 2026Updated 2 weeks ago
Alternatives and similar repositories for Spectrax
Users that are interested in Spectrax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- easydel jax kernels writen in triton for gpus and pallas for tpus☆30Aug 15, 2026Updated 2 weeks ago
- A repo based on XiLin Li's PSGD repo that extends some of the experiments.☆14Oct 7, 2024Updated last year
- A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/…☆34Mar 4, 2025Updated last year
- Implementation of PSGD optimizer in JAX☆37Dec 31, 2024Updated last year
- Xerxes, a highly advanced Persian AI assistant developed by InstinctAI, a cutting-edge AI startup. primary function is to assist users wi…☆15Apr 27, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Automatically take good care of your preemptible TPUs☆37May 15, 2023Updated 3 years ago
- Train vision models using JAX and 🤗 transformers☆104Dec 14, 2025Updated 8 months ago
- OST Collection: An AI-powered suite of models that predict the next word matches with remarkable accuracy (Text Generative Models). OST C…☆16Nov 16, 2023Updated 2 years ago
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆83Aug 13, 2026Updated 2 weeks ago
- This module defines a type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosys…☆36Updated this week
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆16Aug 1, 2022Updated 4 years ago
- Agents for intelligence and coordination☆27Updated this week
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 2024Updated 2 years ago
- Microbenchmarking hyperparameter tuning for JAX functions.☆24Aug 24, 2026Updated last week
- 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.
- A Python DSL to write Nvidia PTX for Hopper and Blackwell in JAX and PyTorch☆377Jul 9, 2026Updated last month
- ☆18Aug 24, 2024Updated 2 years ago
- ☆19Updated this week
- ☆29Updated this week
- ☆25Dec 16, 2025Updated 8 months ago
- ☆34Jun 3, 2024Updated 2 years ago
- ☆60Updated this week
- torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…☆238Aug 7, 2026Updated 3 weeks ago
- ☆23Jan 23, 2024Updated 2 years 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.
- Implementation of N-Grammer in Flax☆17Nov 3, 2022Updated 3 years ago
- ☆121Aug 22, 2026Updated last week
- a Jax quantization library☆131Updated this week
- DALLE-tools provided useful dataset utilities to improve you workflow with WebDatasets.☆14Mar 9, 2022Updated 4 years ago
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated 11 months ago
- ☆591Jul 11, 2024Updated 2 years ago
- ☆269Updated this week
- Minimal yet performant LLM examples in pure JAX☆277Aug 23, 2026Updated last week
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- An implementation of PSGD Kron second-order optimizer for PyTorch☆104Jul 24, 2025Updated last year
- ☆29Jan 8, 2026Updated 7 months ago
- Regionalized Value State Dependence Graph written in Rust☆12Nov 17, 2019Updated 6 years ago
- minimal compiler☆24Feb 19, 2026Updated 6 months ago
- ☆10Apr 5, 2024Updated 2 years ago
- 4-bit Shampoo for Memory-Efficient Network Training (NeurIPS 2024)☆13Feb 13, 2025Updated last year
- JAX implementation of the Llama 2 model☆216Feb 2, 2024Updated 2 years ago