Data compression in JAX

a Python toolkit for fine-tuning Geospatial Foundation Models (GFMs).

This repo contains the code for 1D tokenizer and generator

Statistical and visual tools for gathering metrics and comparing Earth System Model data files. A common use case is comparing data that has been lossily compressed with the original data.

Play atmospheric modelling like it's LEGO.

Memray is a memory profiler for Python

Convection classification with machine learning using triplet trainer

Flax is a neural network library for JAX that is designed for flexibility.

The Fast Cross-Platform Package Manager

Hybrid ML + physics model of the Earth's atmosphere

Automated Bayesian model discovery for time series data

State Space Models library in JAX

An experimental language for causal reasoning

Solve puzzles. Learn CUDA.

Gaussian processes in JAX.

Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods

Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design

Generating AI-ready datasets in python for air quality and EO-related studies using Google Earth Engine

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

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

A debugging and profiling tool that can trace and visualize python code execution

Simulation-based inference benchmark

The TinyLlama project is an open endeavor to pretrain a 1.1B Llama model on 3 trillion tokens.

Code for the paper <SelfCheck: Using LLMs to Zero-Shot Check Their Own Step-by-Step Reasoning>

A domain-specific probabilistic programming language for modeling and inference with language models

Obsidian plugin to automatically create a note from arXiv.org, acl anthology and semantic scholar.

A Library for Gaussian Processes in Chemistry

Easy emulating of geophysical models including (but not limited to!) Earth System Models.