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

• kasper0406 / stablehlo-coreml
  Convert StableHLO models into Apple Core ML format
  ☆19Updated this week
• openxla / community
  Stores documents and resources used by the OpenXLA developer community
  ☆131Updated last year
• albanD / subclass_zoo
  ☆185Updated last year
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆133Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆91Updated 8 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆367Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆84Updated last week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆436Updated last week
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆691Updated this week
• openxla / shardy
  MLIR-based partitioning system
  ☆151Updated this week
• ROCm / iris
  AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming
  ☆133Updated this week
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆117Updated this week
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆579Updated this week
• google / aqt
  ☆340Updated last week
• meta-pytorch / tlparse
  TORCH_LOGS parser for PT2
  ☆69Updated last month
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆60Updated last month
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆71Updated 4 years ago
• iree-org / iree-nvgpu
  ☆50Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆320Updated this week
• bertmaher / simplegemm
  ☆127Updated last month
• meta-pytorch / BackendBench
  Ship correct and fast LLM kernels to PyTorch
  ☆126Updated this week
• Jokeren / triton-samples
  ☆28Updated 11 months ago
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆318Updated 2 weeks ago
• intel / intel-extension-for-openxla
  ☆52Updated last month
• tensorflow / mlir-hlo
  ☆422Updated this week
• yifuwang / symm-mem-recipes
  ☆151Updated 11 months ago
• intel / sycl-tla
  SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs
  ☆59Updated this week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆222Updated this week
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆277Updated 5 years ago
• NVIDIA / nsight-python
  Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools
  ☆71Updated this week