Torch-native, batchable, atomistic simulations.

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

The open source coding agent.

🌎 🤝 AI weather models united

S2FFT: Differentiable and accelerated spherical transforms

Bundle containing all the repositories that are needed to compile JEDI for use in Skylab

Find and load data from the Brightband AI-ready mirror of the NOAA NASA Joint Archive (NNJA) of Observations for Earth System Reanalysis

Benchmarking of machine learning and numerical weather prediction (MLWP & NWP) models, with a focus on extreme events.

Open-source, cloud-native transactional tensor storage engine

🌦️ A catalogue and categorization of AI-based weather forecasting models.

Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)

The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery 🧑‍🔬

Recipes for reproducing Analysis-Ready & Cloud Optimized (ARCO) ERA5 datasets.

Fork of E3SM used to develop exascale global atmosphere model written in C++ (aka SCREAM)

An extensible benchmark suite to evaluate data-driven physical simulation

PDEBench: An Extensive Benchmark for Scientific Machine Learning

Cubed-Sphere data processing with xarray

Explanation to key concepts in ML

A python-based linear inverse modeling suite

Vim notebook cell bindings for JupyterLab

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

A markup-based typesetting system that is powerful and easy to learn.

AutoGPT is the vision of accessible AI for everyone, to use and to build on. Our mission is to provide the tools, so that you can focus on what matters.

A differentiable PDE solving framework for machine learning

Example configurations for MOM6 and SIS2

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.