pytorch/extension-cpp

C++ extensions in PyTorch

What it solves

It provides reference implementations for creating custom C++/CUDA operations in PyTorch, allowing developers to implement high-performance kernels that can run on both CPU and GPU.

How it works

The repository contains two different approaches to building extensions:

  • Standard ATen API: The traditional method using the full ATen/LibTorch API.
  • Stable ABI: A method using the LibTorch Stable ABI (for PyTorch 2.10+), which allows extensions to be compatible across multiple PyTorch versions without requiring recompilation.

Who it’s for

Developers who need to write custom, high-performance operators for PyTorch and want to see how to implement them using either the standard API or the stable ABI.

Highlights

  • Demonstrates a mymuladd custom operation.
  • Includes both CPU and CUDA kernels.
  • Supports the LibTorch Stable ABI for cross-version compatibility.

Related

  • Project
    googleapis/go-genai

    A Go SDK that provides an interface for integrating Google's Gemini generative AI models and the Gemini Enterprise Agent Platform into Go applications.

  • Project
    Sec-Link/Argus-Agentic-SOC-Platform

    Argus is an open‑source, AI‑native Security Operations Center platform that bundles alert ingestion, ticketing, asset inventory, correlation rules, dashboards, and workflow automation. It adds a conversational AI assistant that can answer natural‑language queries, retrieve similar cases, and invoke tools via a JSON‑RPC‑style interface. Built with a Next.js frontend, Django/DRF backend, PostgreSQL (optional Elasticsearch), it offers a modular, API‑first architecture for extending SOC capabilities.

  • Project
    scverse/gget

    A command-line tool and Python package for efficient, single-line querying of genomic reference databases and sequence analysis.

  • Project
    trhhubsd/ARTEXX

    An AI-driven autonomous penetration testing system that uses a multi-agent architecture and dual-graph state management to automate vulnerability discovery and attack chain execution.