🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

A toolkit for developing and comparing reinforcement learning algorithms.

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

You like pytorch? You like micrograd? You love tinygrad! ❤️

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Download market data from Yahoo! Finance's API

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

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

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangChain. Store, query, version, & visualize any AI data. Stream data in real-time to PyTorch/TensorFlow. https://activelo…

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

An elegant PyTorch deep reinforcement learning library.

UltiSnips - The ultimate snippet solution for Vim. Send pull requests to SirVer/ultisnips!

Matplotlib styles for scientific plotting

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Web interface for browsing, search and filtering recent arxiv submissions

my_first_calculator.py

min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch

A library of reinforcement learning components and agents

Check out the new game server:

Awesome work on hand pose estimation/tracking

Python library for loading and using triangular meshes.

The project is an official implement of our ECCV2018 paper "Simple Baselines for Human Pose Estimation and Tracking(https://arxiv.org/abs/1804.06208)"

Code for visualizing the loss landscape of neural nets

View model summaries in PyTorch!

MuZero

A chess library for Python, with move generation and validation, PGN parsing and writing, Polyglot opening book reading, Gaviota tablebase probing, Syzygy tablebase probing, and UCI/XBoard engine c…

Elegant PyTorch implementation of paper Model-Agnostic Meta-Learning (MAML)