Skip to content
Change the repository type filter

All

    Repositories list

    • An extension of `torchvision` for decoding AVIF and HEIC images.
      C++
      GNU Lesser General Public License v2.1
      1621Updated Jun 8, 2025Jun 8, 2025
    • helion

      Public
      A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
      Python
      BSD 3-Clause "New" or "Revised" License
      12155154Updated Jun 8, 2025Jun 8, 2025
    • monarch

      Public
      PyTorch Single Controller
      Rust
      BSD 3-Clause "New" or "Revised" License
      21143331Updated Jun 8, 2025Jun 8, 2025
    • TritonParse is a tool designed to help developers analyze and debug Triton kernels by visualizing the compilation process and source code mappings.
      TypeScript
      BSD 3-Clause "New" or "Revised" License
      11501Updated Jun 6, 2025Jun 6, 2025
    • Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
      Python
      BSD 3-Clause "New" or "Revised" License
      20130126Updated Jun 6, 2025Jun 6, 2025
    • Chakra et_replay is a tool designed for replaying Chakra Execution Traces (ET) from machine learning models using PyTorch.
      Python
      Apache License 2.0
      0100Updated Jun 6, 2025Jun 6, 2025
    • Example apps and demos using PyTorch's ExecuTorch framework
      Swift
      Other
      3930Updated Jun 5, 2025Jun 5, 2025
    • C++ implementations for various tokenizers (sentencepiece, tiktoken etc).
      C++
      BSD 3-Clause "New" or "Revised" License
      73034Updated Jun 5, 2025Jun 5, 2025
    • Helpful tools and examples for working with flex-attention
      Python
      BSD 3-Clause "New" or "Revised" License
      50820603Updated Jun 5, 2025Jun 5, 2025
    • Other
      4000Updated Jun 4, 2025Jun 4, 2025
    • Applied AI experiments and examples for PyTorch
      Python
      BSD 3-Clause "New" or "Revised" License
      28274112Updated May 29, 2025May 29, 2025
    • FACTO

      Public
      Framework for Algorithmic Correctness Testing of Operators
      Python
      BSD 3-Clause "New" or "Revised" License
      2300Updated May 27, 2025May 27, 2025
    • Staging ground for pytorch/pytorch maintainers to collaborate on release notes for PyTorch releases. Houses the artifacts that chronicle each release.
      BSD 3-Clause "New" or "Revised" License
      0000Updated Apr 23, 2025Apr 23, 2025
    • gpt-fast

      Public
      Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
      Python
      BSD 3-Clause "New" or "Revised" License
      5566k7242Updated Apr 11, 2025Apr 11, 2025
    • An experimental CPU backend for Triton (https//github.com/openai/triton)
      C++
      MIT License
      24210Updated Mar 17, 2025Mar 17, 2025
    • torchfix

      Public
      TorchFix - a linter for PyTorch-using code with autofix support
      Python
      Other
      20142136Updated Feb 7, 2025Feb 7, 2025
    • This is a mirror of https//git.mlplatform.org/tosa/serialization_lib.git/
      C++
      Apache License 2.0
      0000Updated Jan 15, 2025Jan 15, 2025
    • This is a mirror of https//git.mlplatform.org/ml/ethos-u/ethos-u-core-driver.git/
      C++
      Apache License 2.0
      0002Updated Nov 21, 2024Nov 21, 2024
    • LeanRL

      Public
      LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
      Python
      Other
      2658462Updated Oct 26, 2024Oct 26, 2024
    • A batched offline inference oriented version of segment-anything
      Python
      Apache License 2.0
      711.2k294Updated Sep 13, 2024Sep 13, 2024
    • superblock

      Public archive
      A block oriented training approach for inference time optimization.
      Python
      MIT License
      43301Updated Aug 19, 2024Aug 19, 2024
    • float8_experimental

      Public archive
      This repository contains the experimental PyTorch native float8 training UX
      Python
      BSD 3-Clause "New" or "Revised" License
      19223022Updated Aug 1, 2024Aug 1, 2024