• Stars
    star
    242
  • Rank 165,711 (Top 4 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created over 2 years ago
  • Updated 2 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

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

Analysis-Ready, Cloud Optimized ERA5

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

IntroductionRoadmapData DescriptionHow to reproduceFAQsHow to cite this workLicense

Introduction

Our goal is to make a global history of the climate highly accessible in the cloud. To that end, we present a curated copy of the ERA5 corpus in Google Cloud Public Datasets.

What is ERA5?

ERA5 is the fifth generation of ECMWF's Atmospheric Reanalysis. It spans atmospheric, land, and ocean variables. ERA5 is an hourly dataset with global coverage at 30km resolution (~0.28° x 0.28°), ranging from 1979 to the present. The total ERA5 dataset is about 5 petabytes in size.

Check out ECMWF's documentation on ERA5 for more.

What is a reanalysis?

A reanalysis is the "most complete picture currently possible of past weather and climate." Reanalyses are created from assimilation of a wide range of data sources via numerical weather prediction (NWP) models.

Read ECMWF's introduction to reanalysis for more.

So far, we have ingested meteorologically valuable variables for the land and atmosphere. From this, we have produced a cloud-optimized version of ERA5, in which we have converted grib data to Zarr with no other modifications. Next, we plan on creating an "analysis-ready" version, oriented towards common research workflows, which we will do in the open.

This two-pronged approach for the data serves different user needs. Some researchers need full control over the interpolation of data for their analysis. Most will want a batteries-included dataset, where standard pre-processing and chunk optimization is already applied. In general, we ensure that every step in this pipeline is open and reproducible, to provide transparency in the provenance of all data.

TODO(#1): What have we done to make this dataset possible?

Please view out our walkthrough notebook for a demo of the datasets.

Roadmap

  1. Phase 0: Ingest raw ERA5
  2. Phase 1: Cloud-Optimize to Zarr, without data modifications
    1. Use Pangeo-Forge to convert the data from grib to Zarr.
    2. Create example notebooks for common workflows, including regridding and variable derivation.
  3. Phase 2: Produce an Analysis-Ready corpus
    1. Regrid datasets to lat/long grids.
    2. Convert model levels to pressure levels (at high resolution).
    3. Compute derived variables.
    4. Expand on example notebooks.
  4. Phase 3: Create an analysis & machine learning (ML) pipeline toolkit
    1. Dataset generator for ML Models.
    2. Examples of reading data in XArray-Beam pipelines.
    3. Notebooks demoing common data analysis, like Extreme Value Analysis.
  5. General future plans...
    1. Include more variables, especially ocean data.
    2. Integrate preliminary ERA5 data (1950 to 1978).
    3. Automatically update with recent data.

Data Description

As of 2022-04-27, all data spans the dates 1979-01-01/to/2021-08-31 (inclusive).

Whenever possible, we have chosen to represent parameters by their native grid resolution. See this ECMWF documentation for more.

Model Level Wind

  • Levels: 1/to/137
  • Times: 00/to/23
  • Grid: Spectral Harmonic Coefficients (docs)
  • Size: 305.89 TiB

Warning: We encountered a small error in one of our source data shards (the dve shard for 2008-08-27 actually had data for 2008-02-27). We noticed this only after ECMWF's MARS archive went down due to a data center migration. In order to release the data as soon as possible, we substituted the previous day's worth of data for this 24-hour period. Once the center is back online, we will re-compute this dataset (#8).

name short name units docs config
vorticity (relative) vo s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=138 era5_ml_dv.cfg
divergence d s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=155 era5_ml_dv.cfg
temperature t K https://apps.ecmwf.int/codes/grib/param-db?id=130 era5_ml_tw.cfg
vertical velocity d Pa s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=135 era5_ml_tw.cfg
import xarray as xr

ml_wind = xr.open_zarr(
    'gs://gcp-public-data-arco-era5/co/model-level-wind.zarr/',
    chunks={'time': 48},
    consolidated=True,
)

Model Level Moisture

name short name units docs config
specific humidity q kg kg^-1 https://apps.ecmwf.int/codes/grib/param-db?id=133 era5_ml_o3q.cfg
ozone mass mixing ratio o3 kg kg^-1 https://apps.ecmwf.int/codes/grib/param-db?id=203 era5_ml_o3q.cfg
specific cloud liquid water content clwc kg kg^-1 https://apps.ecmwf.int/codes/grib/param-db?id=246 era5_ml_o3q.cfg
specific cloud ice water content ciwc kg kg^-1 https://apps.ecmwf.int/codes/grib/param-db?id=247 era5_ml_o3q.cfg
fraction of cloud cover cc (0 - 1) https://apps.ecmwf.int/codes/grib/param-db?id=248 era5_ml_o3q.cfg
specific rain water content crwc kg kg^-1 https://apps.ecmwf.int/codes/grib/param-db?id=75 era5_ml_qrqs.cfg
specific snow water content cswc kg kg^-1 https://apps.ecmwf.int/codes/grib/param-db?id=76 era5_ml_qrqs.cfg
import xarray as xr

ml_moisture = xr.open_zarr(
    'gs://gcp-public-data-arco-era5/co/model-level-moisture.zarr/',
    chunks={'time': 48},
    consolidated=True,
)

Single Level Surface

  • Times: 00/to/23
  • Grid: Spectral Harmonic Coefficients (docs)
  • Size: 1.11 TiB
name short name units docs config
logarithm of surface pressure lnsp ~ https://apps.ecmwf.int/codes/grib/param-db?id=152
surface geopotential zs m^2 s^-2 https://apps.ecmwf.int/codes/grib/param-db?id=162051
import xarray as xr

ml_surface = xr.open_zarr(
    'gs://gcp-public-data-arco-era5/co/single-level-surface.zarr/',
    chunks={'time': 48},
    consolidated=True,
)

Single Level Reanalysis

name short name units docs config
convective available potential energy cape J kg^-1 https://apps.ecmwf.int/codes/grib/param-db?id=59 era5_sfc_cape.cfg
total column cloud ice water tciw kg m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=79 era5_sfc_cape.cfg
vertical integral of divergence of cloud frozen water flux wiiwd kg m^-2 s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=162080 era5_sfc_cape.cfg
100 metre U wind component 100u m s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=228246 era5_sfc_cape.cfg
100 metre V wind component 100v m s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=228247 era5_sfc_cape.cfg
sea ice area fraction ci (0 - 1) https://apps.ecmwf.int/codes/grib/param-db?id=31 era5_sfc_cisst.cfg
sea surface temperature sst Pa https://apps.ecmwf.int/codes/grib/param-db?id=34 era5_sfc_cisst.cfg
skin temperature skt K https://apps.ecmwf.int/codes/grib/param-db?id=235 era5_sfc_cisst.cfg
soil temperature level 1 stl1 K https://apps.ecmwf.int/codes/grib/param-db?id=139 era5_sfc_soil.cfg
soil temperature level 2 stl2 K https://apps.ecmwf.int/codes/grib/param-db?id=170 era5_sfc_soil.cfg
soil temperature level 3 stl3 K https://apps.ecmwf.int/codes/grib/param-db?id=183 era5_sfc_soil.cfg
soil temperature level 4 stl4 K https://apps.ecmwf.int/codes/grib/param-db?id=236 era5_sfc_soil.cfg
temperature of snow layer tsn K https://apps.ecmwf.int/codes/grib/param-db?id=238 era5_sfc_soil.cfg
volumetric soil water layer 1 swvl1 m^3 m^-3 https://apps.ecmwf.int/codes/grib/param-db?id=39 era5_sfc_soil.cfg
volumetric soil water layer 2 swvl2 m^3 m^-3 https://apps.ecmwf.int/codes/grib/param-db?id=40 era5_sfc_soil.cfg
volumetric soil water layer 3 swvl3 m^3 m^-3 https://apps.ecmwf.int/codes/grib/param-db?id=41 era5_sfc_soil.cfg
volumetric soil water layer 4 swvl4 m^3 m^-3 https://apps.ecmwf.int/codes/grib/param-db?id=42 era5_sfc_soil.cfg
ice temperature layer 1 istl1 K https://apps.ecmwf.int/codes/grib/param-db?id=35 era5_sfc_soil.cfg
ice temperature layer 2 istl2 K https://apps.ecmwf.int/codes/grib/param-db?id=36 era5_sfc_soil.cfg
ice temperature layer 3 istl3 K https://apps.ecmwf.int/codes/grib/param-db?id=37 era5_sfc_soil.cfg
ice temperature layer 4 istl4 K https://apps.ecmwf.int/codes/grib/param-db?id=38 era5_sfc_soil.cfg
total column cloud liquid water tclw kg m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=78 era5_sfc_tcol.cfg
total column rain water tcrw kg m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=228089 era5_sfc_tcol.cfg
total column snow water tcsw kg m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=228090 era5_sfc_tcol.cfg
total column water tcw kg m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=136 era5_sfc_tcol.cfg
total column vertically-integrated water vapour tcwv kg m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=137 era5_sfc_tcol.cfg
import xarray as xr

sl_reanalysis = xr.open_zarr(
    'gs://gcp-public-data-arco-era5/co/single-level-reanalysis.zarr', 
    chunks={'time': 48},
    consolidated=True,
)

Single Level Forecast

  • Times: 06:00/18:00
  • Steps: 0/1/2/3/4/5/6/7/8/9/10/11/12/13/14/15/16/17/18
  • Grid: N320, a Reduced Gaussian Grid (docs)
  • Size: 24.52 TiB
name short name units docs config
snow density rsn kg m^-3 https://apps.ecmwf.int/codes/grib/param-db?id=33 era5_sfc_pcp.cfg
snow evaporation es m of water equivalent https://apps.ecmwf.int/codes/grib/param-db?id=44 era5_sfc_pcp.cfg
snow melt smlt m of water equivalent https://apps.ecmwf.int/codes/grib/param-db?id=45 era5_sfc_pcp.cfg
large-scale precipitation fraction lspf s https://apps.ecmwf.int/codes/grib/param-db?id=50 era5_sfc_pcp.cfg
snow depth sd m of water equivalent https://apps.ecmwf.int/codes/grib/param-db?id=141 era5_sfc_pcp.cfg
large-scale precipitation lsp m https://apps.ecmwf.int/codes/grib/param-db?id=142 era5_sfc_pcp.cfg
convective precipitation cp m https://apps.ecmwf.int/codes/grib/param-db?id=143 era5_sfc_pcp.cfg
snowfall sf m of water equivalent https://apps.ecmwf.int/codes/grib/param-db?id=144 era5_sfc_pcp.cfg
convective rain rate crr kg m^-2 s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=228218 era5_sfc_pcp.cfg
large scale rain rate lsrr kg m^-2 s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=228219 era5_sfc_pcp.cfg
convective snowfall rate water equivalent csfr kg m^-2 s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=228220 era5_sfc_pcp.cfg
large scale snowfall rate water equivalent lssfr kg m^-2 s^-1 https://apps.ecmwf.int/codes/grib/param-db?id=228221 era5_sfc_pcp.cfg
total precipitation tp m https://apps.ecmwf.int/codes/grib/param-db?id=228 era5_sfc_pcp.cfg
convective snowfall csf m of water equivalent https://apps.ecmwf.int/codes/grib/param-db?id=239 era5_sfc_pcp.cfg
large-scale snowfall lsf m of water equivalent https://apps.ecmwf.int/codes/grib/param-db?id=240 era5_sfc_pcp.cfg
precipitation type ptype code table (4.201) https://apps.ecmwf.int/codes/grib/param-db?id=260015 era5_sfc_pcp.cfg
surface solar radiation downwards ssrd J m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=169 era5_sfc_rad.cfg
top net thermal radiation ttr J m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=179 era5_sfc_rad.cfg
gravity wave dissipation gwd J m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=197 era5_sfc_rad.cfg
surface thermal radiation downwards strd J m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=175 era5_sfc_rad.cfg
surface net thermal radiation str J m^-2 https://apps.ecmwf.int/codes/grib/param-db?id=177 era5_sfc_rad.cfg
import xarray as xr

sl_forecasts = xr.open_zarr(
    'gs://gcp-public-data-arco-era5/co/single-level-forecast.zarr/', 
    chunks={'time': 48},
    consolidated=True,
)

How to reproduce

All phases of this dataset can be reproduced with scripts found here. To run them, please clone the repo and install the project.

git clone https://github.com/google-research/arco-era5.git

Or, via SSH:

git clone [email protected]:google-research/arco-era5.git

Then, install with pip:

cd arco-era5
pip install -e .

Acquire & preprocess raw data

Please consult the instructions described in raw/.

Cloud-Optimization

All our tools make use of Apache Beam, and thus are portable to any cloud (or Runner). We use GCP's Dataflow to produce this dataset. If you would like to reproduce the project this way, we recommend the following:

  1. Ensure you have access to a GCP project with GCS read & write access, as well as full Dataflow permissions (see these "Before you begin" instructions).
  2. Export the following variables:
    export PROJECT=<your-gcp-project>
    export REGION=us-central1
    export BUCKET=<your-beam-runner-bucket>

From here, we provide examples of how to run the recipes at the top of each script.

pydoc src/single-levels-to-zarr.py

You can also discover available command line options by invoking the script with -h/--help:

python src/model-levels-to-zarr.py --help

Making the dataset "Analysis Ready" & beyond...

This phase of the project is under active development! If you would like to lend a hand in any way, please check out our contributing guide.

FAQs

How did you pick these variables?

This dataset originated in Loon, Alphabet’s project to deliver internet service using stratospheric balloons, and is now curated by Google Research & Google Cloud Platform. Loon’s Planning, Simulation and Control team needed accurate data on how the stratospheric winds have behaved in the past to evaluate the effectiveness of different balloon steering algorithms over a range of weather. This led us to download the model-level data. But Loon also needed information about the atmospheric radiation to model balloon gas temperatures, so we downloaded that. And then we downloaded the most commonly used meteorological variables to support different product planning needs (RF propagation models, etc)...

Eventually, we found ourselves with a comprehensive history of weather for the world.

Where are the U/V components of wind? Where is geopotential height? Why isn’t X variable in this dataset?

We intentionally did not include many variables that can be derived from other variables. For example, U/V components of wind can be computed from divergence and vorticity; geopotential is a vertical integral of temperature.

In the second phase of our roadmap (towards "Analysis Ready" data), we aim to compute all of these variables ourselves. If you’d like to make use of these parameters sooner, please check out our example notebooks where we demo common calculations. If you notice non-derived missing data, such as surface variables, please let us know of your needs by filing an issue, and we will be happy to incorporate them into our roadmap.

Do you have plans to get all of ERA5?

We aim to support hosting data that serves general meteorological use cases, rather than aim for total completeness. Wave variables are missing from this corpus, and are a priority on our roadmap. If there is a variable or dataset that you think should be included here, please file a Github issue.

For a complete ERA5 mirror, we recommend consulting with the Pangeo Forge project (especially staged-recipes#92).

Why are there two model-level datasets and not one?

It definitely is possible for all model level data to be represented in one grid, and thus one dataset. However, we opted to preserve the native representation for variables in ECMWF's models. A handful of core model variables (wind, temperature and surface pressure) are represented as spectral harmonic coefficients , while everything else is stored on a Gaussian grid. This avoids introducing numerical error by interpolating these variables to physical space. For a more in depth review of this topic, please consult these references:

Please note: in a future releases, we intend to create a dataset version where all model levels are in one grid and Zarr.

Why doesn’t this project make use of Pangeo Forge Cloud?

We are big fans of the Pangeo Forge project, and of Pangeo in general. While this project does make use of their Recipes, we have a few reasons to not use their cloud. First, we would prefer to use internal rather than community resources for computations of this scale. In addition, there are several technical reasons why Pangeo Forge as it is today would not be able to handle this case (0, 1, 2, 3). To work around this, we opted to combine familiar-to-us infrastructure with Pangeo-Forge's core and to use the right tool for the right job.

Why use this dataset? What uses are there for the data?

ERA5 can be used in many applications. It can be used to train ML models that predict the impact of weather on different phenomena. ERA5 data could also be used to train and evaluate ML models that forecast the weather. The data could be used to compute climatologies, or the average weather for a region over a given period of time. ERA5 data can be used to visualize and study historical weather events, such as Hurricane Sandy.

Where should I be cautious? What are the limitations of the dataset?

Mumbai, India
Mumbai, India
San Francisco, USA
San Francisco, USA
Tokyo, Japan
Tokyo, Japan
Singapore
Singapore
ERA5 Topography
ERA5 Topography
GMTED2010 Topography
GMTED2010 Topography

It is important to remember that a reanalysis is an estimate of what the weather was, it is not guaranteed to be an error-free estimate. There are several areas where the novice reanalysis user should be careful.

First, the user should be careful using reanalysis data at locations near coastlines. The first figure shows the fraction of land (1 for land, 0 for ocean) of ERA5 grid points at different coastal locations. This is important because the land-surface model used in ERA5 tries to blend in the influence of water with the influence of land based on this fraction. The most visible effect of this blending is that as the fraction of land decreases, the daily variation in temperature will also decrease. Looking at the first figure, there are sharp changes in the fraction of land between neighboring grid cells so there could be differences in daily temperature range that might not be reflected in actual weather observations.

The user should also be careful when using reanalysis data in areas with large variations in topography. The second figure is a plot of ERA5 topography around Mount Everest compared with GMTED2010 topography. The ERA5 topography is completely missing the high peaks of the Everest region and missing most of the structure of the mountain valleys. Topography strongly influences temperature and precipitation rate, so it is possible that ERA5’s temperature is too warm and ERA5’s precipitation patterns could be wrong as well.

ERA5’s precipitation variables aren’t directly constrained by any observations, so we strongly encourage the user to check ERA5 against observed precipitation (for example, Wu et al., 2022). A study comparing reanalyses (not including ERA5) against gridded precipitation observations showed striking differences between reanalyses and observation Lisa V Alexander et al 2020 Environ. Res. Lett. 15 055002.

Can I use the data for {research,commercial} purposes?

Yes, you can use our ERA5 data according to the terms of the Copernicus license.

Researchers, see the next section for how to cite this work.

Commercial users, please be sure to provide acknowledgement to the Copernicus Climate Change Service according to the Copernicus Licence terms.

How to cite this work

Please cite our presentation at the 22nd Conference on Artificial Intelligence for Environmental Science describing ARCO-ERA5.

Carver, Robert W, and Merose, Alex. (2023): ARCO-ERA5: An Analysis-Ready Cloud-Optimized Reanalysis Dataset. 22nd Conf. on AI for Env. Science, Denver, CO, Amer. Meteo. Soc, 4A.1, https://ams.confex.com/ams/103ANNUAL/meetingapp.cgi/Paper/415842

In addition, please cite the ERA5 dataset accordingly:

Hersbach, H., Bell, B., Berrisford, P., Hirahara, S., Horányi, A., 
Muñoz‐Sabater, J., Nicolas, J., Peubey, C., Radu, R., Schepers, D., 
Simmons, A., Soci, C., Abdalla, S., Abellan, X., Balsamo, G., 
Bechtold, P., Biavati, G., Bidlot, J., Bonavita, M., De Chiara, G., 
Dahlgren, P., Dee, D., Diamantakis, M., Dragani, R., Flemming, J., 
Forbes, R., Fuentes, M., Geer, A., Haimberger, L., Healy, S., 
Hogan, R.J., Hólm, E., Janisková, M., Keeley, S., Laloyaux, P., 
Lopez, P., Lupu, C., Radnoti, G., de Rosnay, P., Rozum, I., Vamborg, F.,
Villaume, S., Thépaut, J-N. (2017): Complete ERA5: Fifth generation of 
ECMWF atmospheric reanalyses of the global climate. Copernicus Climate 
Change Service (C3S) Data Store (CDS). (Accessed on DD-MM-YYYY)

Hersbach et al, (2017) was downloaded from the Copernicus Climate Change 
Service (C3S) Climate Data Store. We thank C3S for allowing us to 
redistribute the data.

The results contain modified Copernicus Climate Change Service 
information 2022. Neither the European Commission nor ECMWF is 
responsible for any use that may be made of the Copernicus information 
or data it contains.

License

This is not an official Google product.

Copyright 2022 Google LLC

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    https://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

More Repositories

1

bert

TensorFlow code and pre-trained models for BERT
Python
37,288
star
2

google-research

Google Research
Jupyter Notebook
33,759
star
3

tuning_playbook

A playbook for systematically maximizing the performance of deep learning models.
26,142
star
4

vision_transformer

Jupyter Notebook
9,944
star
5

text-to-text-transfer-transformer

Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"
Python
6,054
star
6

arxiv-latex-cleaner

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv
Python
5,004
star
7

simclr

SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners
Jupyter Notebook
3,937
star
8

multinerf

A Code Release for Mip-NeRF 360, Ref-NeRF, and RawNeRF
Python
3,579
star
9

timesfm

TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.
Python
3,469
star
10

football

Check out the new game server:
Python
3,260
star
11

albert

ALBERT: A Lite BERT for Self-supervised Learning of Language Representations
Python
3,209
star
12

scenic

Scenic: A Jax Library for Computer Vision Research and Beyond
Python
3,192
star
13

frame-interpolation

FILM: Frame Interpolation for Large Motion, In ECCV 2022.
Python
2,729
star
14

t5x

Python
2,622
star
15

electra

ELECTRA: Pre-training Text Encoders as Discriminators Rather Than Generators
Python
2,308
star
16

kubric

A data generation pipeline for creating semi-realistic synthetic multi-object videos with rich annotations such as instance segmentation masks, depth maps, and optical flow.
Jupyter Notebook
2,273
star
17

big_vision

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.
Jupyter Notebook
2,136
star
18

uda

Unsupervised Data Augmentation (UDA)
Python
2,131
star
19

pegasus

Python
1,600
star
20

language

Shared repository for open-sourced projects from the Google AI Language team.
Python
1,598
star
21

dex-lang

Research language for array processing in the Haskell/ML family
Haskell
1,580
star
22

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
Python
1,543
star
23

parti

1,527
star
24

big_transfer

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
Python
1,500
star
25

FLAN

Python
1,446
star
26

disentanglement_lib

disentanglement_lib is an open-source library for research on learning disentangled representations.
Python
1,311
star
27

robotics_transformer

Python
1,260
star
28

multilingual-t5

Python
1,197
star
29

planet

Learning Latent Dynamics for Planning from Pixels
Python
1,134
star
30

mixmatch

Python
1,130
star
31

tapas

End-to-end neural table-text understanding models.
Python
1,127
star
32

deduplicate-text-datasets

Rust
1,087
star
33

fixmatch

A simple method to perform semi-supervised learning with limited data.
Python
1,079
star
34

morph-net

Fast & Simple Resource-Constrained Learning of Deep Network Structure
Python
1,013
star
35

deeplab2

DeepLab2 is a TensorFlow library for deep labeling, aiming to provide a unified and state-of-the-art TensorFlow codebase for dense pixel labeling tasks.
Python
995
star
36

maxim

[CVPR 2022 Oral] Official repository for "MAXIM: Multi-Axis MLP for Image Processing". SOTA for denoising, deblurring, deraining, dehazing, and enhancement.
Python
971
star
37

batch-ppo

Efficient Batched Reinforcement Learning in TensorFlow
Python
963
star
38

augmix

AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty
Python
951
star
39

magvit

Official JAX implementation of MAGVIT: Masked Generative Video Transformer
Python
913
star
40

pix2seq

Pix2Seq codebase: multi-tasks with generative modeling (autoregressive and diffusion)
Jupyter Notebook
843
star
41

seed_rl

SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's architecture.
Python
792
star
42

meta-dataset

A dataset of datasets for learning to learn from few examples
Jupyter Notebook
752
star
43

noisystudent

Code for Noisy Student Training. https://arxiv.org/abs/1911.04252
Python
740
star
44

recsim

A Configurable Recommender Systems Simulation Platform
Python
729
star
45

rliable

[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.
Jupyter Notebook
728
star
46

jax3d

Python
727
star
47

circuit_training

Python
716
star
48

long-range-arena

Long Range Arena for Benchmarking Efficient Transformers
Python
707
star
49

lottery-ticket-hypothesis

A reimplementation of "The Lottery Ticket Hypothesis" (Frankle and Carbin) on MNIST.
Python
705
star
50

federated

A collection of Google research projects related to Federated Learning and Federated Analytics.
Python
662
star
51

bleurt

BLEURT is a metric for Natural Language Generation based on transfer learning.
Python
651
star
52

nasbench

NASBench: A Neural Architecture Search Dataset and Benchmark
Python
641
star
53

prompt-tuning

Original Implementation of Prompt Tuning from Lester, et al, 2021
Python
632
star
54

xtreme

XTREME is a benchmark for the evaluation of the cross-lingual generalization ability of pre-trained multilingual models that covers 40 typologically diverse languages and includes nine tasks.
Python
626
star
55

lasertagger

Python
606
star
56

sound-separation

Python
603
star
57

dreamer

Dream to Control: Learning Behaviors by Latent Imagination
Python
568
star
58

pix2struct

Python
568
star
59

fast-soft-sort

Fast Differentiable Sorting and Ranking
Python
561
star
60

robopianist

[CoRL '23] Dexterous piano playing with deep reinforcement learning.
Python
550
star
61

ravens

Train robotic agents to learn pick and place with deep learning for vision-based manipulation in PyBullet. Transporter Nets, CoRL 2020.
Python
546
star
62

sam

Python
539
star
63

vmoe

Jupyter Notebook
534
star
64

batch_rl

Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games
Python
521
star
65

bigbird

Transformers for Longer Sequences
Python
518
star
66

tensor2robot

Distributed machine learning infrastructure for large-scale robotics research
Python
483
star
67

byt5

Python
477
star
68

adapter-bert

Python
470
star
69

omniglue

Code release for CVPR'24 submission 'OmniGlue'
Python
469
star
70

mint

Multi-modal Content Creation Model Training Infrastructure including the FACT model (AI Choreographer) implementation.
Python
465
star
71

leaf-audio

LEAF is a learnable alternative to audio features such as mel-filterbanks, that can be initialized as an approximation of mel-filterbanks, and then be trained for the task at hand, while using a very small number of parameters.
Python
446
star
72

robustness_metrics

Jupyter Notebook
442
star
73

receptive_field

Compute receptive fields of your favorite convnets
Python
428
star
74

maxvit

[ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmentation, image quality, and generative modeling...
Jupyter Notebook
425
star
75

ssl_detection

Semi-supervised learning for object detection
Python
398
star
76

maskgit

Official Jax Implementation of MaskGIT
Jupyter Notebook
394
star
77

l2p

Learning to Prompt (L2P) for Continual Learning @ CVPR22 and DualPrompt: Complementary Prompting for Rehearsal-free Continual Learning @ ECCV22
Python
388
star
78

nerf-from-image

Shape, Pose, and Appearance from a Single Image via Bootstrapped Radiance Field Inversion
Python
373
star
79

computation-thru-dynamics

Understanding computation in artificial and biological recurrent networks through the lens of dynamical systems.
Jupyter Notebook
369
star
80

tf-slim

Python
368
star
81

distilling-step-by-step

Python
366
star
82

weatherbench2

A benchmark for the next generation of data-driven global weather models.
Python
360
star
83

realworldrl_suite

Real-World RL Benchmark Suite
Python
341
star
84

python-graphs

A static analysis library for computing graph representations of Python programs suitable for use with graph neural networks.
Python
325
star
85

rigl

End-to-end training of sparse deep neural networks with little-to-no performance loss.
Python
314
star
86

self-organising-systems

Jupyter Notebook
306
star
87

task_adaptation

Python
305
star
88

tensorflow_constrained_optimization

Python
300
star
89

ibc

Official implementation of Implicit Behavioral Cloning, as described in our CoRL 2021 paper, see more at https://implicitbc.github.io/
Python
295
star
90

syn-rep-learn

Learning from synthetic data - code and models
Python
286
star
91

exoplanet-ml

Machine learning models and utilities for exoplanet science.
Python
283
star
92

vdm

Jupyter Notebook
280
star
93

retvec

RETVec is an efficient, multilingual, and adversarially-robust text vectorizer.
Jupyter Notebook
277
star
94

tensorflow-coder

Python
275
star
95

lm-extraction-benchmark

Python
266
star
96

sparf

This is the official code release for SPARF: Neural Radiance Fields from Sparse and Noisy Poses [CVPR 2023-Highlight]
Python
264
star
97

falken

Falken provides developers with a service that allows them to train AI that can play their games
Python
253
star
98

rlds

Jupyter Notebook
252
star
99

3d-moments

Code for CVPR 2022 paper '3D Moments from Near-Duplicate Photos'
Python
236
star
100

meliad

Python
234
star