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
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆380Updated this week
• albanD / subclass_zoo
  ☆189Updated last year
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆64Updated 2 weeks ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆88Updated last week
• openxla / shardy
  MLIR-based partitioning system
  ☆164Updated last week
• cchan / tccl
  extensible collectives library in triton
  ☆95Updated 10 months ago
• kasper0406 / stablehlo-coreml
  Convert StableHLO models into Apple Core ML format
  ☆21Updated last week
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆118Updated last week
• ROCm / iris
  AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming
  ☆164Updated last week
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆601Updated 3 weeks ago
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆739Updated this week
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆73Updated 4 years ago
• intel / intel-extension-for-openxla
  ☆55Updated 2 months ago
• meta-pytorch / tlparse
  TORCH_TRACE parser for PT2
  ☆75Updated last week
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆169Updated last week
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆138Updated 8 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆328Updated last week
• Jokeren / triton-samples
  ☆28Updated last year
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆437Updated last month
• meta-pytorch / tritonparse
  TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels
  ☆194Updated this week
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆325Updated 2 months ago
• triton-lang / kernels
  ☆104Updated last year
• iree-org / iree-nvgpu
  ☆49Updated last year
• nod-ai / AMD-SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆104Updated last month
• yifuwang / symm-mem-recipes
  ☆159Updated last year
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆315Updated 5 months ago
• google / aqt
  ☆345Updated last week
• iree-org / iree-turbine
  IREE's PyTorch Frontend, based on Torch Dynamo.
  ☆105Updated last week
• csarofeen / pytorch
  Tensors and Dynamic neural networks in Python with strong GPU acceleration
  ☆26Updated 2 years ago