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
  ☆126Updated last year
• openxla / shardy
  MLIR-based partitioning system
  ☆117Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆75Updated last week
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆346Updated this week
• albanD / subclass_zoo
  ☆171Updated last year
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆52Updated last month
• kasper0406 / stablehlo-coreml
  Convert StableHLO models into Apple Core ML format
  ☆19Updated 2 weeks ago
• cchan / tccl
  extensible collectives library in triton
  ☆88Updated 4 months ago
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆70Updated 4 years ago
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆212Updated this week
• intel / intel-extension-for-openxla
  ☆49Updated 2 months ago
• pytorch / tlparse
  TORCH_LOGS parser for PT2
  ☆48Updated this week
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆110Updated this week
• bertmaher / simplegemm
  ☆111Updated 4 months ago
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆412Updated last month
• iree-org / iree-nvgpu
  ☆50Updated last year
• intel / torch-xpu-ops
  ☆50Updated this week
• nod-ai / SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆101Updated this week
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆517Updated this week
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆264Updated this week
• manishucsd / py-codegen
  ☆16Updated 10 months ago
• iree-org / iree-jax
  ☆52Updated last year
• tensorflow / mlir-hlo
  ☆420Updated this week
• google / aqt
  ☆324Updated last week
• triton-lang / triton-cpu
  An experimental CPU backend for Triton
  ☆139Updated 2 months ago
• makslevental / mlir-python-extras
  The missing pieces (as far as boilerplate reduction goes) of the upstream MLIR python bindings.
  ☆104Updated this week
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆311Updated 4 months ago
• Jokeren / triton-samples
  ☆28Updated 6 months ago
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆80Updated last year
• spcl / pymlir
  Python interface for MLIR - the Multi-Level Intermediate Representation
  ☆263Updated 8 months ago