Google Research

Top repositories by stars

• google-research/bert(on leaderboard)

  TensorFlow code and pre-trained models for BERT

  Python39,861
• google-research/google-research(on leaderboard)

  Google Research

  Jupyter Notebook37,271
• google-research/tuning_playbook

  A playbook for systematically maximizing the performance of deep learning models.

  29,812
• google-research/vision_transformer

  Jupyter Notebook12,302
• google-research/timesfm

  TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.

  Python7,751
• google-research/arxiv-latex-cleaner

  arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

  Python6,707
• google-research/text-to-text-transfer-transformer

  Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"

  Python6,489
• google-research/simclr

  SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners

  Jupyter Notebook4,453
• google-research/multinerf

  A Code Release for Mip-NeRF 360, Ref-NeRF, and RawNeRF

  Python3,795
• google-research/scenic

  Scenic: A Jax Library for Computer Vision Research and Beyond

  Python3,766
• google-research/football

  Check out the new game server:

  Python3,550
• google-research/big_vision

  Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

  Jupyter Notebook3,361
• google-research/albert

  ALBERT: A Lite BERT for Self-supervised Learning of Language Representations

  Python3,276
• google-research/frame-interpolation

  FILM: Frame Interpolation for Large Motion, In ECCV 2022.

  Python3,108
• google-research/t5x

  Python2,947
• google-research/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 Notebook2,669
• google-research/electra

  ELECTRA: Pre-training Text Encoders as Discriminators Rather Than Generators

  Python2,371
• google-research/uda

  Unsupervised Data Augmentation (UDA)

  Python2,204
• google-research/language

  Shared repository for open-sourced projects from the Google AI Language team.

  Python1,747
• google-research/torchsde

  Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

  Python1,705
• google-research/robotics_transformer

  Python1,678
• google-research/dex-lang

  Research language for array processing in the Haskell/ML family

  Haskell1,661
• google-research/pegasus

  Python1,651
• google-research/circuit_training

  Python1,595
• google-research/parti

  1,590
• google-research/FLAN

  Python1,560
• google-research/big_transfer

  Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

  Python1,539
• google-research/disentanglement_lib

  disentanglement_lib is an open-source library for research on learning disentangled representations.

  Python1,420
• google-research/multilingual-t5

  Python1,296
• google-research/deduplicate-text-datasets

  Rust1,257
• google-research/planet

  Learning Latent Dynamics for Planning from Pixels

  Python1,234
• google-research/tapas

  End-to-end neural table-text understanding models.

  Python1,203
• google-research/fixmatch

  A simple method to perform semi-supervised learning with limited data.

  Python1,203
• google-research/mixmatch

  Python1,145
• google-research/maxim

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

  Python1,084
• google-research/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.

  Python1,033
• google-research/morph-net

  Fast & Simple Resource-Constrained Learning of Deep Network Structure

  Python1,031
• google-research/magvit

  Official JAX implementation of MAGVIT: Masked Generative Video Transformer

  Python993
• google-research/augmix

  AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty

  Python989
• google-research/batch-ppo

  Efficient Batched Reinforcement Learning in TensorFlow

  Python974
• google-research/inksight

  Jupyter Notebook948
• google-research/pix2seq

  Pix2Seq codebase: multi-tasks with generative modeling (autoregressive and diffusion)

  Jupyter Notebook939
• google-research/rliable

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

  Jupyter Notebook866
• google-research/seed_rl

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

  Python837
• google-research/meta-dataset

  A dataset of datasets for learning to learn from few examples

  Jupyter Notebook798
• google-research/bleurt

  BLEURT is a metric for Natural Language Generation based on transfer learning.

  Python784
• google-research/recsim

  A Configurable Recommender Systems Simulation Platform

  Python782
• google-research/long-range-arena

  Long Range Arena for Benchmarking Efficient Transformers

  Python777
• google-research/noisystudent

  Code for Noisy Student Training. https://arxiv.org/abs/1911.04252

  Python763
• google-research/jax3d

  Python758
• google-research/federated

  A collection of Google research projects related to Federated Learning and Federated Analytics.

  Python742
• google-research/lottery-ticket-hypothesis

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

  Python725
• google-research/nasbench

  NASBench: A Neural Architecture Search Dataset and Benchmark

  Python711
• google-research/sound-separation

  Python707
• google-research/dreamer

  Dream to Control: Learning Behaviors by Latent Imagination

  Python706
• google-research/vmoe

  Jupyter Notebook705
• google-research/robopianist

  [CoRL '23] Dexterous piano playing with deep reinforcement learning.

  Python703
• google-research/prompt-tuning

  Original Implementation of Prompt Tuning from Lester, et al, 2021

  Python698
• google-research/omniglue

  Code release for CVPR'24 submission 'OmniGlue'

  Python696
• google-research/pix2struct

  Python679
• google-research/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.

  Python650
• google-research/bigbird

  Transformers for Longer Sequences

  Python631
• google-research/android_world

  AndroidWorld is an environment and benchmark for autonomous agents

  Python630
• google-research/ravens

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

  Python621
• google-research/sam

  Python620
• google-research/fast-soft-sort

  Fast Differentiable Sorting and Ranking

  Python616
• google-research/lasertagger

  Python604
• google-research/weatherbench2

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

  Python591
• google-research/distilling-step-by-step

  Python580
• google-research/tensor2robot

  Distributed machine learning infrastructure for large-scale robotics research

  Python563
• google-research/batch_rl

  Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games

  Python559
• google-research/maskgit

  Official Jax Implementation of MaskGIT

  Jupyter Notebook554
• google-research/mint

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

  Python551
• google-research/byt5

  Python534
• google-research/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.

  Python520
• google-research/adapter-bert

  Python505
• google-research/maxvit

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

  Jupyter Notebook488
• google-research/l2p

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

  Python472
• google-research/robustness_metrics

  Jupyter Notebook472
• google-research/rlds

  Jupyter Notebook463
• google-research/receptive_field

  Compute receptive fields of your favorite convnets

  Python449
• google-research/arco-era5

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

  Python444
• google-research/ssl_detection

  Semi-supervised learning for object detection

  Python414
• google-research/self-organising-systems

  Jupyter Notebook397
• google-research/nerf-from-image

  Shape, Pose, and Appearance from a Single Image via Bootstrapped Radiance Field Inversion

  Python395
• google-research/computation-thru-dynamics

  Understanding computation in artificial and biological recurrent networks through the lens of dynamical systems.

  Jupyter Notebook389
• google-research/population-dynamics

  PDFM Embeddings: location-based vectors for geo-spatial analysis.

  Jupyter Notebook380
• google-research/tf-slim

  Python372
• google-research/ibc

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

  Python371
• google-research/realworldrl_suite

  Real-World RL Benchmark Suite

  Python363
• google-research/language-table

  Suite of human-collected datasets and a multi-task continuous control benchmark for open vocabulary visuolinguomotor learning.

  Jupyter Notebook348
• google-research/task_adaptation

  Python341
• google-research/python-graphs

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

  Python340
• google-research/rigl

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

  Python335
• google-research/vdm

  Jupyter Notebook328
• google-research/syn-rep-learn

  Learning from synthetic data - code and models

  Python327
• google-research/Splat-SLAM

  Splat-SLAM: Globally Optimized RGB-only SLAM with 3D Gaussians

  Python326
• google-research/exoplanet-ml

  Machine learning models and utilities for exoplanet science.

  Python317
• google-research/perch

  Python312
• google-research/tensorflow_constrained_optimization

  Python312