glouppe

grok-1

Grok open release

InstantID

InstantID : Zero-shot Identity-Preserving Generation in Seconds 🔥

graphcast

jax-md

Differentiable, Hardware Accelerated, Molecular Dynamics

Pangu-Weather

An official implementation of Pangu-Weather

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

ferminet

An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations

PDEBench

PDEBench: An Extensive Benchmark for Scientific Machine Learning

FourCastNet

Initial public release of code, data, and model weights for FourCastNet

AIRS

Artificial Intelligence Research for Science (AIRS)

weatherbench2

A benchmark for the next generation of data-driven global weather models.

ai-models

metnet

PyTorch Implementation of Google Research's MetNet and MetNet-2

einx

Universal Tensor Operations in Einstein-Inspired Notation for Python.

Score-Entropy-Discrete-Diffusion

[ICML 2024 Oral] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)

dinosaur

scalax

A simple library for scaling up JAX programs

Awesome-LWMs

🌍 A Collection of Awesome Large Weather Models (LWMs) | AI for Earth (AI4Earth) | AI for Science (AI4Science)

ai-models-panguweather

FengWu

ai-models-graphcast

AstroCLIP

Multimodal contrastive pretraining for astronomical data

GNOT

ai-models-fourcastnetv2

ai-models-fuxi

IP_and_Im_Lectures

10 Lectures on Inverse Problems and Imaging

inox

Stainless neural networks in JAX

PAPERCLIP-Hubble

Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/2403.08851.

lc2st

Offical code for "L-C2ST: Local Validation Diagnostics for Posterior Approximations in Simulation-Based Inference", NeurIPS 2023