Tensors and Dynamic neural networks in Python with strong GPU acceleration

Animation engine for explanatory math videos

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Optional static typing for Python

The interactive graphing library for Python ✨

A computer algebra system written in pure Python

The pytest framework makes it easy to write small tests, yet scales to support complex functional testing

Open Machine Learning Compiler Framework

Style transfer, deep learning, feature transform

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

A Python framework for accelerated simulation, data generation and spatial computing.

Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels

A Python library that generates static type annotations by collecting runtime types

System for AI Education Resource.

This is Meta's fork of the CPython runtime. The name "cinder" here is historical, see https://github.com/facebookincubator/cinderx for the Python extension / JIT compiler.

High-Performance Symbolic Regression in Python and Julia

Official pytorch implementation of the paper: "SinGAN: Learning a Generative Model from a Single Natural Image"

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

Multithreaded Python without the GIL

TensorFlow Graphics: Differentiable Graphics Layers for TensorFlow

Source-to-Source Debuggable Derivatives in Pure Python

Simple script for downloading Youtube comments without using the Youtube API

ChatDBG - AI-assisted debugging. Uses AI to answer 'why'

Reinforcement learning environments for compiler and program optimization tasks

Official Code Release for [SIGGRAPH 2025] RenderFormer: Transformer-based Neural Rendering of Triangle Meshes with Global Illumination

A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.

Tensorflow (Python API) implementation of Deep Photo Style Transfer

Learning infinite-resolution image processing with GAN and RL from unpaired image datasets, using a differentiable photo editing model.

Infrastructure for Machine Learning Guided Optimization (MLGO) in LLVM.

Advanced Physics Engines 2020: A Hands-on Tutorial