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

• albanD / subclass_zoo
  ☆184Updated last year
• openxla / community
  Stores documents and resources used by the OpenXLA developer community
  ☆131Updated last year
• kasper0406 / stablehlo-coreml
  Convert StableHLO models into Apple Core ML format
  ☆19Updated last week
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆363Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆91Updated 8 months ago
• openxla / shardy
  MLIR-based partitioning system
  ☆150Updated this week
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆57Updated last week
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆115Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆84Updated 2 weeks ago
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆640Updated this week
• ROCm / iris
  AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming
  ☆116Updated last week
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆400Updated 2 months ago
• intel / intel-extension-for-openxla
  ☆51Updated last week
• iree-org / iree-nvgpu
  ☆50Updated last year
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆573Updated last week
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆71Updated 4 years ago
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆116Updated this week
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆437Updated last week
• yifuwang / symm-mem-recipes
  ☆147Updated 11 months ago
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆436Updated this week
• meta-pytorch / tlparse
  TORCH_LOGS parser for PT2
  ☆65Updated 2 weeks ago
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆136Updated 6 months ago
• bertmaher / simplegemm
  ☆126Updated last month
• meta-pytorch / BackendBench
  Ship correct and fast LLM kernels to PyTorch
  ☆124Updated 2 weeks ago
• Jokeren / triton-samples
  ☆28Updated 10 months ago
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆307Updated 3 months ago
• Dao-AILab / quack
  A Quirky Assortment of CuTe Kernels
  ☆675Updated last week
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆312Updated this week
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆313Updated last month
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆151Updated 2 years ago