google-research-datasets / tpu_graphsLinks
☆129Updated 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)☆112Updated last year
- Collection of kernels written in Triton language☆173Updated 8 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆216Updated this week
- ☆338Updated last week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆161Updated 2 months ago
- Write a fast kernel and run it on Discord. See how you compare against the best!☆64Updated 2 weeks ago
- 🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.☆64Updated 10 months ago
- NeurIPS Large Language Model Efficiency Challenge: 1 LLM + 1GPU + 1Day☆258Updated 2 years ago
- ☆252Updated last year
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆153Updated 2 years ago
- ☆83Updated 2 years ago
- ☆115Updated last year
- Training material for IPU users: tutorials, feature examples, simple applications☆87Updated 2 years ago
- A list of awesome neural symbolic papers.☆50Updated 3 years ago
- ML model training for edge devices☆167Updated 2 years ago
- Code for studying the super weight in LLM☆121Updated last year
- ring-attention experiments☆160Updated last year
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆404Updated this week
- Cataloging released Triton kernels.☆277Updated 3 months ago
- ☆28Updated 10 months ago
- ☆262Updated this week
- Experiments on Multi-Head Latent Attention☆99Updated last year
- A Python library transfers PyTorch tensors between CPU and NVMe☆122Updated last year
- A schedule language for large model training☆151Updated 3 months ago
- Home for OctoML PyTorch Profiler☆114Updated 2 years ago
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- Ship correct and fast LLM kernels to PyTorch☆126Updated this week
- ☆47Updated last year
- [ICLR'25] Fast Inference of MoE Models with CPU-GPU Orchestration☆246Updated last year
- Research and development for optimizing transformers☆131Updated 4 years ago