tvm
https://github.com/apache/tvm
Python
Open deep learning compiler stack for cpu, gpu and specialized accelerators
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported3 Subscribers
Add a CodeTriage badge to tvm
Help out
- Issues
- [REFACTOR][CUDA] Clarify CUDA codegen flow
- [TOPI] Use branchless boundary index for reflect/replicate pad
- [Bug] [Relax] `InternalError: Divide by zero` when compiling a `reshape` that yields a 0-sized dimension
- [TIRx][Schedule] Support rfactor for arg reducers selecting last index
- [TOPI][CUDA] Fix topk/sort gridDim overflow by remapping grid axes in sort_ir
- [Relax][VM] Improve diagnostics for unlowered Relax operators
- [Relax][ONNX] Add CastLike support and dynamic-k Trilu to expand backend coverage
- [Relax][Frontend][TFLite] Support StableHLO shape ops
- # [Performance] `nn.pad` reflect/replicate ~1.5x slower than needed since v0.21 (nested `if_then_else` boundary index)
- [Runtime][WebGPU] Add prerequisites for Gemma 4 E2B WebGPU execution
- Docs
- Python not yet supported