Stars
A modular framework for evaluating weather forecasts
A comprehensive list of published machine learning applications to cosmology
Differentiable and GPU accelerated scattering covariance statistics on the sphere
10 differentiable physical simulators built with Taichi differentiable programming (DiffTaichi, ICLR 2020)
[ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
PyTorch implementation of the Hessian-free optimizer
Implementation of the Aurora model for Earth system forecasting
bonevbs / torch-harmonics
Forked from NVIDIA/torch-harmonicsDifferentiable Spherical Harmonic transforms in PyTorch
Official inference repo for FLUX.1 models
Schedule-Free Optimization in PyTorch
Codomain attention neural operator for single to multi-physics PDE adaptation.
GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection
Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.
Recipes for reproducing Analysis-Ready & Cloud Optimized (ARCO) ERA5 datasets.
A benchmark for the next generation of data-driven global weather models.
This repository is the official implementation of the paper Convolutional Neural Operators for robust and accurate learning of PDEs
Machine Learning Engineering Open Book
Yet another PyTorch implementation of Stable Diffusion (probably easy to read)
LLMs as Copilots for Theorem Proving in Lean
Fork of E3SM used to develop exascale global atmosphere model written in C++ (aka SCREAM)
Run AI NWP forecasts hassle-free, serverless in the cloud!
Windowed alignment of time series at the speed of light
A beautiful, simple, clean, and responsive Jekyll theme for academics
12 weeks, 26 lessons, 52 quizzes, classic Machine Learning for all