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

• openxla / community
  Stores documents and resources used by the OpenXLA developer community
  ☆129Updated last year
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆76Updated last week
• albanD / subclass_zoo
  ☆176Updated last year
• cchan / tccl
  extensible collectives library in triton
  ☆87Updated 5 months ago
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆53Updated last month
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆70Updated 4 years ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆353Updated this week
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆541Updated this week
• openxla / shardy
  MLIR-based partitioning system
  ☆135Updated this week
• pytorch / tlparse
  TORCH_LOGS parser for PT2
  ☆60Updated this week
• simveit / effective_transpose
  Effective transpose on Hopper GPU
  ☆23Updated 2 weeks ago
• iree-org / iree-nvgpu
  ☆50Updated last year
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆111Updated this week
• kasper0406 / stablehlo-coreml
  Convert StableHLO models into Apple Core ML format
  ☆19Updated last month
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆307Updated this week
• facebookexperimental / triton
  Github mirror of trition-lang/triton repo.
  ☆75Updated this week
• Jokeren / triton-samples
  ☆28Updated 8 months ago
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆346Updated last week
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆80Updated last year
• bertmaher / simplegemm
  ☆118Updated 6 months ago
• nod-ai / SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆101Updated last month
• intel / intel-extension-for-openxla
  ☆51Updated 3 months ago
• manishucsd / py-codegen
  ☆16Updated last year
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆41Updated last year
• triton-lang / kernels
  ☆90Updated 10 months ago
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆423Updated 3 weeks ago
• ColfaxResearch / layout-categories
  This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".
  ☆29Updated this week
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆287Updated 3 weeks ago
• yifuwang / symm-mem-recipes
  ☆119Updated 8 months ago
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆223Updated last week