AdamCobb

pyprob

A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch

PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.

virtualhome

API to run VirtualHome, a Multi-Agent Household Simulator

openshs

Open Smart Home Simulator

RELLIS-3D

RELLIS-3D: A Multi-modal Dataset for Off-Road Robotics

awesome-ml-courses

Awesome free machine learning and AI courses with video lectures.

open-data

Free football data from StatsBomb

DALLE-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

wilds

A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.

pymanopt

Python toolbox for optimization on Riemannian manifolds with support for automatic differentiation

uncertainty-baselines

High-quality implementations of standard and SOTA methods on a variety of tasks.

TyXe

Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)

cvxpylayers

Differentiable convex optimization layers

GalSim

The modular galaxy image simulation toolkit. Documentation:

particles

Sequential Monte Carlo in python

open_lth

A repository in preparation for open-sourcing lottery ticket hypothesis code.

CSCS-summer-school-2017

CSCS-ICS-DADSi Summer School: Accelerating Data Science with HPC, September 4 – 6, 2017

rebound

💫 An open-source multi-purpose N-body code.

ai-safety-gridworlds

This is a suite of reinforcement learning environments illustrating various safety properties of intelligent agents.

transformers

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

financial-machine-learning

A curated list of practical financial machine learning tools and applications.

shrinkbench

PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.

blitz-bayesian-deep-learning

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.

brevitas

Brevitas: neural network quantization in PyTorch

AirSim

Open source simulator for autonomous vehicles built on Unreal Engine / Unity, from Microsoft AI & Research

awesome-multimodal-ml

Reading list for research topics in multimodal machine learning

dnn-mode-connectivity

Mode Connectivity and Fast Geometric Ensembles in PyTorch

benchmark

minGPT

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

tdw

ThreeDWorld simulation environment