Autograd Python Packages

torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

autograd

Efficiently computes derivatives of NumPy code.

autograd-gamma

NotImplementedError: VJP of gammainc wrt argnum 0 not defined

pennylane

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

torch-sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

lucid-dl

Lumerico's Comprehensive Interface for Deep Learning

neuralengine

An open-source framework/library for building and training neural networks in Python with support for GPUs (CUDA).

neuralfabric

From tensors to transformers, built from scratch.

ntk-ml

A from-scratch neural network and tensor library built in Python for learning, experimentation, and deep understanding of modern ML systems.

datarax

A Differentiable Data Pipeline Framework for JAX

eiko

Eiko: the GPU-accelerated Eikonal equation solver

pyerrors

Error propagation and statistical analysis for Monte Carlo simulations in lattice QCD and statistical mechanics using autograd.

mygrad

Drop-in autodiff for NumPy.

norse

Deep learning with spiking neural networks (SNNs) in PyTorch.

nabla-ml

Nabla: High-Performance Scientific Computing

agent-torch

large population models

nnetflow

A minimal, NumPy-based deep learning library with a clean, educational reverse-mode autodiff engine

legume-gme

🌱 Guided-mode expansion of photonic crystal slabs

nutens

Tensor based engine for calculating neutrino oscillation probabilities in a fast, flexible, and differentiable way

pyember

ML Library from scratch with only standard libraries

boost-loss

Utilities for easy use of custom losses in CatBoost, LightGBM, XGBoost.

nets

Recreating PyTorch from scratch, using Numpy. Supports FCN, CNN, RNN layers.

torchvine

Pure-PyTorch vine copula modelling — GPU-ready, differentiable, and API-compatible with pyvinecopulib

essos

e-Stellarator Simulation and Optimization Suite