Tile primitives for speedy kernels

Machine Learning Engineering Open Book

Grok open release

Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper

Massively parallel rigidbody physics simulation on accelerator hardware.

A library of reinforcement learning components and agents

Sparse nonlinear least squares in JAX

100 exercises to learn JAX

Examples of how to create colorful, annotated equations in Latex using Tikz.

CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image

Python 3.8+ toolbox for submitting jobs to Slurm

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Minimax Optimization, Stackelberg Games, Generative Adversarial Networks

Differentiable convex optimization layers

This repository is no longer maintained. Check

Bayesian optimization in PyTorch

Matplotlib 3.1 cheat sheet.

code for "Residual Flows for Invertible Generative Modeling".

Deep neural network kernel for Gaussian process

Efficient and Diverse Batch Acquisition for Deep Bayesian Active Learning.

Web-based dashboard for Sacred

Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.

Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088

Code for reproducing experiments in Model-Based Active Exploration, ICML 2019

Collection of recent methods on (deep) neural network compression and acceleration.

Regularization, Neural Network Training Dynamics

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".

Adaptive Experimentation Platform