albanD / pytorch_dev_env_setup

Related projects ⓘ

Alternatives and complementary repositories for pytorch_dev_env_setup

• openxla / community
  Stores documents and resources used by the OpenXLA developer community
  ☆106Updated 3 months ago
• openxla / shardy
  MLIR-based partitioning system
  ☆36Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆65Updated last month
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆268Updated this week
• iree-org / iree-jax
  ☆48Updated 3 months ago
• albanD / subclass_zoo
  ☆148Updated 4 months ago
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆78Updated this week
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆67Updated 3 years ago
• nod-ai / SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆95Updated this week
• ROCm / rocMLIR
  ☆128Updated this week
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆80Updated last year
• iree-org / iree-turbine
  IREE's PyTorch Frontend, based on Torch Dynamo.
  ☆53Updated this week
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆188Updated this week
• mmperf / mmperf
  MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.
  ☆123Updated last year
• triton-lang / triton-cpu
  An experimental CPU backend for Triton
  ☆56Updated this week
• spcl / pymlir
  Python interface for MLIR - the Multi-Level Intermediate Representation
  ☆224Updated 3 weeks ago
• makslevental / mlir-python-extras
  The missing pieces (as far as boilerplate reduction goes) of the upstream MLIR python bindings.
  ☆66Updated last week
• google / aqt
  ☆266Updated this week
• plaidml / tpp-mlir
  TPP experimentation on MLIR for linear algebra
  ☆111Updated 3 weeks ago
• bwasti / pytorch_compiler_tutorial
  Codebase associated with the PyTorch compiler tutorial
  ☆44Updated 5 years ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆210Updated this week
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆128Updated 2 months ago
• intel / mlir-extensions
  Intel® Extension for MLIR. A staging ground for MLIR dialects and tools for Intel devices using the MLIR toolchain.
  ☆123Updated this week
• daadaada / turingas
  Assembler for NVIDIA Volta and Turing GPUs
  ☆200Updated 2 years ago
• numba / numba-mlir
  POC work on MLIR backend
  ☆48Updated 2 months ago
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆143Updated this week
• csarofeen / pytorch
  Tensors and Dynamic neural networks in Python with strong GPU acceleration
  ☆26Updated last year
• triton-lang / kernels
  ☆43Updated this week
• uwplse / tensat
  Re-implementation of the TASO compiler using equality saturation
  ☆121Updated 3 years ago