google-research-datasets / tpu_graphsLinks
☆125Updated last year
Alternatives and similar repositories for tpu_graphs
Users that are interested in tpu_graphs are comparing it to the libraries listed below
Sorting:
- Memory Optimizations for Deep Learning (ICML 2023)☆108Updated last year
- Training material for IPU users: tutorials, feature examples, simple applications☆87Updated 2 years ago
- Collection of kernels written in Triton language☆157Updated 6 months ago
- ring-attention experiments☆152Updated 11 months ago
- ☆28Updated 8 months ago
- ☆113Updated last year
- A list of awesome neural symbolic papers.☆48Updated 3 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆161Updated 3 weeks ago
- Write a fast kernel and run it on Discord. See how you compare against the best!☆58Updated 2 weeks ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆143Updated last year
- ☆252Updated last year
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆213Updated last week
- Official repository of Sparse ISO-FLOP Transformations for Maximizing Training Efficiency☆25Updated last year
- NeurIPS Large Language Model Efficiency Challenge: 1 LLM + 1GPU + 1Day☆256Updated last year
- [ICLR'25] Fast Inference of MoE Models with CPU-GPU Orchestration☆235Updated 10 months ago
- Sparsity support for PyTorch☆37Updated 6 months ago
- 🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.☆64Updated 8 months ago
- A schedule language for large model training☆151Updated last month
- ☆333Updated 3 weeks ago
- Fast low-bit matmul kernels in Triton☆376Updated last week
- Cataloging released Triton kernels.☆261Updated last month
- A Python library transfers PyTorch tensors between CPU and NVMe☆121Updated 10 months ago
- ☆242Updated last week
- This repository contains the experimental PyTorch native float8 training UX☆223Updated last year
- ML model training for edge devices☆168Updated 2 years ago
- Triton-based implementation of Sparse Mixture of Experts.☆242Updated last week
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆68Updated 10 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆301Updated this week
- ☆173Updated last year
- Parallel framework for training and fine-tuning deep neural networks☆64Updated 6 months ago