AI-Hypercomputer / jetstream-pytorch
PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
☆60Updated last month
Alternatives and similar repositories for jetstream-pytorch:
Users that are interested in jetstream-pytorch are comparing it to the libraries listed below
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆322Updated this week
- Google TPU optimizations for transformers models☆109Updated 3 months ago
- ☆138Updated this week
- Applied AI experiments and examples for PyTorch☆262Updated last week
- Fast low-bit matmul kernels in Triton☆295Updated this week
- ☆25Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 9 months ago
- ☆104Updated 8 months ago
- extensible collectives library in triton☆85Updated last month
- Collection of kernels written in Triton language☆120Updated last month
- ring-attention experiments☆132Updated 6 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆244Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆120Updated this week
- ☆186Updated last week
- ☆68Updated last month
- Boosting 4-bit inference kernels with 2:4 Sparsity☆73Updated 8 months ago
- ☆297Updated this week
- Triton-based implementation of Sparse Mixture of Experts.☆212Updated 5 months ago
- Write a fast kernel and run it on Discord. See how you compare against the best!☆44Updated this week
- A safetensors extension to efficiently store sparse quantized tensors on disk☆102Updated this week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆194Updated this week
- ☆202Updated last week
- A tool to configure, launch and manage your machine learning experiments.☆144Updated this week
- ☆209Updated 3 months ago
- Load compute kernels from the Hub☆115Updated last week
- ☆126Updated 2 months ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆126Updated 5 months ago
- PyTorch per step fault tolerance (actively under development)☆291Updated this week
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)☆71Updated this week