fbartolic

tuning_playbook

A playbook for systematically maximizing the performance of deep learning models.

d2l-en

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

The-Art-of-Linear-Algebra

Graphic notes on Gilbert Strang's "Linear Algebra for Everyone"

Awesome-Diffusion-Models

A collection of resources and papers on Diffusion Models

Tensor-Puzzles

Solve puzzles. Improve your pytorch.

uncertainty-toolbox

Uncertainty Toolbox: a Python toolbox for predictive uncertainty quantification, calibration, metrics, and visualization

vizier

Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.

fortuna

A Library for Uncertainty Quantification.

chex

WeatherBench

A benchmark dataset for data-driven weather forecasting

ClimaX

Foundation model for weather & climate

distrax

modulus-makani

Massively parallel training of machine-learning based weather and climate models

dinosaur

metnet3-pytorch

Implementation of MetNet-3, SOTA neural weather model out of Google Deepmind, in Pytorch

properscoring

Proper scoring rules in Python

s2fft

Differentiable and accelerated spherical transforms with JAX

spherical-cnn

quax

Multiple dispatch over abstract array types in JAX.

DLWP-CS

Deep learning models for global weather prediction on a cubed sphere

Pigeons.jl

Sampling from intractable distributions, with support for distributed and parallel methods

quadax

Numerical quadrature with JAX

Equivariant-Net

model-diagnostics

Tools for diagnostics and assessment of (machine learning) models

quantile-aggregation

Code for Flexible Model Aggregation for Quantile Regression

ensemble_transformer

Official PyTorch implementation of "Self-Attentive Ensemble Transformer: Representing Ensemble Interactions in Neural Networks for Earth System Models".

rbig_jax

Iterative and Parametric Gaussianization with JAX.

gaussflow

Gaussianization Flows for density estimation, sampling and information theory measures (RBIG2.0)

ghq

Gauss-Hermite Quadrature

Bigsimr.jl

Simulate multivariate distributions with arbitrary marginals.