AI-Hypercomputer / cloud-accelerator-diagnostics
☆15Updated last month
Related projects ⓘ
Alternatives and complementary repositories for cloud-accelerator-diagnostics
- ☆223Updated 4 months ago
- ☆178Updated last week
- jax-triton contains integrations between JAX and OpenAI Triton☆343Updated last week
- ☆266Updated last week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆457Updated 2 weeks ago
- ☆64Updated 2 years ago
- A simple library for scaling up JAX programs☆125Updated last week
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆80Updated this week
- extensible collectives library in triton☆65Updated last month
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆40Updated this week
- ☆197Updated 3 months ago
- JAX Synergistic Memory Inspector☆164Updated 3 months ago
- Accelerated First Order Parallel Associative Scan☆162Updated 2 months ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆300Updated 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…☆229Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆211Updated 3 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆516Updated this week
- ☆155Updated last week
- seqax = sequence modeling + JAX☆132Updated 3 months ago
- JAX-Toolbox☆242Updated this week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆210Updated this week
- Google TPU optimizations for transformers models☆74Updated last week
- ☆300Updated last week
- ☆145Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆191Updated 3 weeks ago
- JAX implementation of the Llama 2 model☆210Updated 9 months ago
- JAX bindings for Flash Attention v2☆79Updated 3 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆98Updated last month
- A tool to configure, launch and manage your machine learning experiments.☆62Updated this week
- Applied AI experiments and examples for PyTorch☆160Updated last week