manishucsd / py-codegenLinks
☆16Updated last year
Alternatives and similar repositories for py-codegen
Users that are interested in py-codegen are comparing it to the libraries listed below
Sorting:
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆102Updated 5 months ago
- ☆50Updated last year
- ☆52Updated 7 months ago
- ☆28Updated 10 months ago
- extensible collectives library in triton☆91Updated 8 months ago
- MLIR-based partitioning system☆151Updated this week
- Ahead of Time (AOT) Triton Math Library☆84Updated 3 weeks ago
- A lightweight, Pythonic, frontend for MLIR☆80Updated 2 years ago
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆81Updated 2 months ago
- ☆23Updated 3 months ago
- ☆96Updated last year
- GEMM and Winograd based convolutions using CUTLASS☆28Updated 5 years ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆123Updated last year
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆47Updated 3 months ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆137Updated 3 years ago
- A library of GPU kernels for sparse matrix operations.☆277Updated 5 years ago
- Sparsity support for PyTorch☆37Updated 8 months ago
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆58Updated 3 weeks ago
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆121Updated this week
- A lightweight triton-based General Matrix Multiplication (GEMM) library.☆28Updated this week
- Code for the paper “Four Over Six: More Accurate NVFP4 Quantization with Adaptive Block Scaling”☆52Updated this week
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆140Updated 2 years ago
- An extension library of WMMA API (Tensor Core API)☆109Updated last year
- Wave: Python Domain-Specific Language for High Performance Machine Learning☆26Updated this week
- TORCH_LOGS parser for PT2☆68Updated 3 weeks ago
- Framework to reduce autotune overhead to zero for well known deployments.☆88Updated 2 months ago
- cuASR: CUDA Algebra for Semirings☆42Updated 3 years ago
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆90Updated 2 years ago
- Effective transpose on Hopper GPU☆27Updated 3 months ago
- ☆20Updated 6 years ago