AI-Hypercomputer / maxdiffusion
☆188Updated this week
Alternatives and similar repositories for maxdiffusion:
Users that are interested in maxdiffusion are comparing it to the libraries listed below
- Google TPU optimizations for transformers models☆98Updated 3 weeks ago
- ☆134Updated 2 weeks ago
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆275Updated this week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆478Updated 2 weeks ago
- Scalable and Performant Data Loading☆217Updated this week
- JAX implementation of the Llama 2 model☆215Updated last year
- ☆182Updated 2 weeks ago
- This repository contains the experimental PyTorch native float8 training UX☆221Updated 6 months ago
- A simple library for scaling up JAX programs☆129Updated 3 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆122Updated 10 months ago
- Focused on fast experimentation and simplicity☆65Updated last month
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆51Updated last week
- ☆75Updated 7 months ago
- ☆211Updated 7 months ago
- Implementation of the Llama architecture with RLHF + Q-learning☆162Updated 2 weeks ago
- JAX-Toolbox☆280Updated this week
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆29Updated 4 months ago
- Efficient optimizers☆169Updated this week
- Implementation of Flash Attention in Jax☆204Updated 11 months ago
- supporting pytorch FSDP for optimizers☆76Updated 2 months ago
- seqax = sequence modeling + JAX☆143Updated 7 months ago
- LoRA for arbitrary JAX models and functions☆135Updated 11 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆379Updated 3 weeks ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆107Updated 3 weeks ago
- PyTorch per step fault tolerance (actively under development)☆243Updated this week
- Experiment of using Tangent to autodiff triton☆75Updated last year
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆103Updated this week
- Inference code for LLaMA models in JAX☆114Updated 9 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆95Updated 3 months ago
- An implementation of PSGD Kron second-order optimizer for PyTorch☆83Updated last week