A collection of megakernels for TPU
☆163Oct 15, 2026Updated this week
Alternatives and similar repositories for tpu-megakernels
Users that are interested in tpu-megakernels are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- An Open Compiler Harness for Agentic GPU Programming☆159Updated this week
- ☆75Sep 30, 2026Updated 2 weeks ago
- ☆145Oct 1, 2026Updated 2 weeks ago
- GPU kernel optimization framework☆32Sep 4, 2026Updated last month
- FLUX 3 Image Klein: reference inference code☆87Updated this week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆53Updated this week
- ☆10Nov 28, 2023Updated 2 years ago
- Small System One decision models (17M–400M) for Yes/No, Choice, Score, and reranking. Training toolkit, Python inference, and ONNX suppor…☆37Updated this week
- FLA but cuTile☆27Apr 17, 2026Updated 6 months ago
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆31Jan 22, 2026Updated 8 months ago
- UEval: A Benchmark for Unified Multimodal Generation☆26Apr 20, 2026Updated 5 months ago
- Click this --> https://zsdonghao.github.io☆10Updated this week
- Official Code Repo for "Learning on the Manifold: Unlocking Standard Diffusion Transformers with Representation Encoders" (ECCV 2026))☆24Jul 15, 2026Updated 3 months ago
- [ICML 2026] The official implementation of "Mean Flow Distillation: Robust and Stable Distillation for Flow Matching Models".☆29Jun 14, 2026Updated 4 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Some improvements on ArcFace model☆17Sep 10, 2022Updated 4 years ago
- 免费的AI视频生成nonebot插件，支持文生视频和图文生视频☆11May 7, 2025Updated last year
- Automate the BMad Dev Cycle☆159Updated this week
- MoonEP: A Perfectly Balanced Expert Parallelism Library via Dynamic Redundant Experts☆1,182Updated this week
- LoopFormer is an elastic-depth looped Transformer trained on variable-length trajectories, using time/step-size conditioning and a shortc…☆52Mar 28, 2026Updated 6 months ago
- A unified inference runtime for VLA models.☆241Updated this week
- Trying to implement https://arxiv.org/abs/2305.08891☆35Jun 10, 2023Updated 3 years ago
- High-performance GPU kernels for Ads and Recsys model training, independently implemented and optimized for real-world workloads and mode…☆57Updated this week
- Koishi's Day 2024 Paper (NeurIPS 2024): An advanced persona-driven role-playing system with global faithfulness quantification and optimi…☆14Oct 19, 2025Updated 11 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆56Sep 27, 2026Updated 2 weeks ago
- ☆12Nov 14, 2024Updated last year
- [Tech Report] Context Scaling: Scaling Properties of Text Conditioning in Visual Generation☆187Aug 7, 2026Updated 2 months ago
- High-performance Rust benchmark client for vLLM serving endpoints.☆51Aug 3, 2026Updated 2 months ago
- A notebook using numpy, scipy, and matplotlib to show the concepts taught in the course Linear Algebra by Prof. Gilbert Strang at MIT☆11Jul 16, 2022Updated 4 years ago
- Open-source evaluation toolkit of large vision-language models (LVLMs), support ~100 VLMs, 30+ benchmarks☆15Feb 17, 2025Updated last year
- PyTorch Code for Explorative Modeling: Unlocking a Third Pretraining Axis and End-to-End Generation☆249Sep 30, 2026Updated 2 weeks ago
- Official Training and Benchmarking Implementation of PDMD: Projected Distribution Matching Distillation for Video Diffusion Models☆75Oct 8, 2026Updated last week
- The agent flow system for token maxxing.☆317Updated this week
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Reference implementation and examples of the CuTe Layout representation and algebra.☆377Sep 30, 2026Updated 2 weeks ago
- ☆40Updated this week
- ☆16Oct 23, 2023Updated 2 years ago
- ☆15Mar 11, 2026Updated 7 months ago
- Quickly setup the infrastructure to run a collaborative autoresearch project☆46Updated this week
- Official Implementation of "DiffusionART: Efficient, Stable and Scalable Reinforcement of Diffusion Models"☆141Updated this week
- A Rust-based Unikernel Enhancing Reliability and Efficiency of Embedded Systems.☆12Jun 28, 2024Updated 2 years ago