kaixih / tf_op_graph
A visualization tool to display TF-Grappler optimized op graph
☆12Updated 2 years ago
Alternatives and similar repositories for tf_op_graph:
Users that are interested in tf_op_graph are comparing it to the libraries listed below
- Standalone Flash Attention v2 kernel without libtorch dependency☆99Updated 4 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆88Updated 11 months ago
- ☆36Updated last month
- ☆38Updated 4 years ago
- ☆15Updated this week
- ☆48Updated 10 months ago
- An extension library of WMMA API (Tensor Core API)☆87Updated 6 months ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆175Updated this week
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆127Updated last year
- ☆67Updated last month
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆77Updated 2 months ago
- ☆84Updated 9 months ago
- ☆180Updated 6 months ago
- An IR for efficiently simulating distributed ML computation.☆25Updated last year
- PyTorch distributed training acceleration framework☆39Updated last week
- FlexFlow Serve: Low-Latency, High-Performance LLM Serving☆17Updated this week
- Experimental projects related to TensorRT☆86Updated this week
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆47Updated this week
- CUTLASS and CuTe Examples☆37Updated 3 weeks ago
- ☆36Updated last month
- CUDA Matrix Multiplication Optimization☆155Updated 6 months ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- OpenAI Triton backend for Intel® GPUs☆157Updated this week
- This is a demo how to write a high performance convolution run on apple silicon☆52Updated 2 years ago
- ☆60Updated last month
- ☆79Updated 4 months ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆64Updated 6 years ago
- ☆64Updated 2 months ago
- An experimental CPU backend for Triton☆81Updated last week
- End to End steps for adding custom ops in PyTorch.☆20Updated 4 years ago