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
  ☆133Updated last year
• albanD / subclass_zoo
  ☆189Updated last year
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆380Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆95Updated 10 months ago
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆169Updated last week
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆64Updated 2 weeks ago
• openxla / shardy
  MLIR-based partitioning system
  ☆164Updated last week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆88Updated last week
• ROCm / iris
  AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming
  ☆168Updated this week
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆739Updated this week
• kasper0406 / stablehlo-coreml
  Convert StableHLO models into Apple Core ML format
  ☆21Updated last week
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆73Updated 4 years ago
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆118Updated last week
• Jokeren / triton-samples
  ☆28Updated last year
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆138Updated 8 months ago
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆601Updated 3 weeks ago
• iree-org / iree-nvgpu
  ☆49Updated last year
• facebookexperimental / triton
  Github mirror of trition-lang/triton repo.
  ☆128Updated this week
• NVIDIA / nsight-python
  Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools
  ☆111Updated this week
• manishucsd / py-codegen
  ☆16Updated last year
• meta-pytorch / tlparse
  TORCH_TRACE parser for PT2
  ☆75Updated last week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆226Updated last week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆437Updated last month
• triton-lang / kernels
  ☆104Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆328Updated last week
• bertmaher / simplegemm
  ☆130Updated 3 months ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆165Updated last year
• iree-org / iree-turbine
  IREE's PyTorch Frontend, based on Torch Dynamo.
  ☆105Updated this week
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆462Updated this week
• nod-ai / AMD-SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆104Updated last month