albanD / pytorch_dev_env_setupLinks
☆11Updated last year
Alternatives and similar repositories for pytorch_dev_env_setup
Users that are interested in pytorch_dev_env_setup are comparing it to the libraries listed below
Sorting:
- Stores documents and resources used by the OpenXLA developer community☆130Updated last year
- ☆177Updated last year
- Ahead of Time (AOT) Triton Math Library☆78Updated last week
- extensible collectives library in triton☆89Updated 6 months ago
- Convert StableHLO models into Apple Core ML format☆19Updated 2 months ago
- TORCH_LOGS parser for PT2☆62Updated 3 weeks ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆357Updated this week
- Convert nvprof profiles into about:tracing compatible JSON files☆70Updated 4 years ago
- Experimental projects related to TensorRT☆113Updated last week
- MLIR-based partitioning system☆137Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆389Updated this week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆54Updated 2 weeks ago
- Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!☆99Updated this week
- ☆50Updated last year
- ☆28Updated 9 months ago
- ☆92Updated 11 months ago
- Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…☆67Updated 2 weeks ago
- jax-triton contains integrations between JAX and OpenAI Triton☆428Updated this week
- ☆332Updated last month
- Backward compatible ML compute opset inspired by HLO/MHLO☆552Updated this week
- Applied AI experiments and examples for PyTorch☆299Updated last month
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆155Updated last week
- Fastest kernels written from scratch☆374Updated 3 weeks ago
- ☆120Updated 7 months ago
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆67Updated 3 weeks ago
- Github mirror of trition-lang/triton repo.☆84Updated last week
- A library to analyze PyTorch traces.☆416Updated 2 weeks ago
- An experimental CPU backend for Triton☆153Updated last week
- High-Performance SGEMM on CUDA devices☆107Updated 8 months ago
- ☆124Updated 9 months ago