AI-Hypercomputer / tpu-recipesLinks
☆33Updated last week
Alternatives and similar repositories for tpu-recipes
Users that are interested in tpu-recipes are comparing it to the libraries listed below
Sorting:
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆60Updated 2 months ago
- Write a fast kernel and run it on Discord. See how you compare against the best!☆44Updated this week
- ☆20Updated last year
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆123Updated this week
- Load compute kernels from the Hub☆139Updated this week
- A place to store reusable transformer components of my own creation or found on the interwebs☆56Updated 3 weeks ago
- ☆138Updated 2 weeks ago
- ☆21Updated 3 months ago
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)☆105Updated this week
- Experiment of using Tangent to autodiff triton☆79Updated last year
- ☆116Updated 2 weeks ago
- ☆46Updated last week
- ☆108Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆110Updated 3 weeks ago
- Train, tune, and infer Bamba model☆127Updated last month
- Learn CUDA with PyTorch☆21Updated this week
- A set of Python scripts that makes your experience on TPU better☆54Updated 11 months ago
- ☆17Updated last year
- Various transformers for FSDP research☆37Updated 2 years ago
- Google TPU optimizations for transformers models☆112Updated 4 months ago
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆93Updated 10 months ago
- PyTorch per step fault tolerance (actively under development)☆302Updated this week
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆335Updated this week
- ☆16Updated 2 months ago
- ☆67Updated 2 years ago
- Minimal but scalable implementation of large language models in JAX☆34Updated 7 months ago
- extensible collectives library in triton☆87Updated 2 months ago
- ☆44Updated last year
- ☆78Updated 11 months ago
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆45Updated 2 weeks ago