Pytest plugin for testing examples in docstrings and markdown files.

Machine Learning with Symbolic Tensors

eXpression differentiation in Julia

Hyper optimized contraction trees for large tensor networks and einsums

TensorLy: Tensor Learning in Python.

Convolutions and more as einsum for PyTorch

a compact tensor library capable of training deep neural networks on both cpu and cuda devices

Simple LaTeX parser providing latex-to-unicode and unicode-to-latex conversion

An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO

The matrix cookbook, proved in the Lean theorem prover

a categorical deep learning compiler

Fuji is an AI agent that lives in your browser's sidepanel. You can now get tasks done online with a single command!

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.

Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditioner and more)

Collection of Matrix/Linear Algebra Information

Uncertainty quantification with PyTorch

Macros in Python: quasiquotes, case classes, LINQ and more!

SGLang is a fast serving framework for large language models and vision language models.

An open-source visual programming environment for battle-testing prompts to LLMs.

CLI tool to fix command line errors with ChatGPT. When you hit an error your command line, get immediate help and fix from ChatGPT.

D2 is a modern diagram scripting language that turns text to diagrams.

Welcome to the Llama Cookbook! This is your go to guide for Building with Llama: Getting started with Inference, Fine-Tuning, RAG. We also show you how to solve end to end problems using Llama mode…

a bridge between python and shell

Set of datasets for the deep learning recommendation model (DLRM).

Named tensors with first-class dimensions for PyTorch

Few shot learning for human pose estimation