Google Research

google-research

Google Research

vision_transformer

scenic

Scenic: A Jax Library for Computer Vision Research and Beyond

t5x

timesfm

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

big_vision

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

dex-lang

Research language for array processing in the Haskell/ML family

jax3d

prompt-tuning

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

vmoe

robustness_metrics

weatherbench2

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

self-organising-systems

perch

dinosaur

spherical-cnn

optformer

neuralgcm

Hybrid ML + physics model of the Earth's atmosphere

jestimator

Amos optimizer with JEstimator lib.

hyperbo

Pre-trained Gaussian processes for Bayesian optimization

structured-additive-IR

dataclass_array

Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)

mozolm

MozoLM: A language model (LM) serving library

kauldron

Modular, scalable codebase to train ML models

nisaba

Finite-state script normalization and processing utilities

swirl-dynamics

precondition

gpax

diffren

swirl-fem