Skip to content

@google-research Google Research

All

    Repositories list

    251 repositories

    • kauldron

      Public
      Modular, scalable library to train ML models
      pythondeep-learningjax
      pythondeep-learningjax
      Python
      •
      Apache License 2.0
      •35•306•7•73•Updated Oct 5, 2026Oct 5, 2026

    • flood-forecasting

      Public
      Python
      •
      Apache License 2.0
      •94•351•12•7•Updated Oct 5, 2026Oct 5, 2026

    • perch-hoplite

      Public
      Tooling for agile modeling on large machine perception embedding databases.
      Python
      •
      Apache License 2.0
      •38•128•14•10•Updated Oct 5, 2026Oct 5, 2026

    • weatherbenchX

      Public
      A modular framework for evaluating weather forecasts
      Python
      •
      Apache License 2.0
      •40•260•18•71•Updated Oct 5, 2026Oct 5, 2026

    • perch

      Public
      Python
      •
      Apache License 2.0
      •70•398•1•16•Updated Oct 5, 2026Oct 5, 2026

    • last

      Public
      A JAX library for building lattice-based speech transducer models
      Python
      •
      Apache License 2.0
      •6•48•0•2•Updated Oct 5, 2026Oct 5, 2026

    • spherical-cnn

      Public
      Python
      •
      Apache License 2.0
      •8•145•5•2•Updated Oct 5, 2026Oct 5, 2026

    • pix2struct

      Public
      Python
      •
      Apache License 2.0
      •62•692•19•8•Updated Oct 5, 2026Oct 5, 2026

    • jax3d

      Public
      Python
      •
      Apache License 2.0
      •93•761•41•8•Updated Oct 5, 2026Oct 5, 2026

    • hyperbo

      Public
      Pre-trained Gaussian processes for Bayesian optimization
      machine-learningdeep-learningbayesian-methods
      machine-learningdeep-learningbayesian-methodsbayesian-optimizationhyperparameter-tuning
      Python
      •
      Apache License 2.0
      •9•105•6•5•Updated Oct 5, 2026Oct 5, 2026

    • swirl-fem

      Public
      Python
      •
      Apache License 2.0
      •2•6•0•1•Updated Oct 5, 2026Oct 5, 2026

    • morph-net

      Public
      Fast & Simple Resource-Constrained Learning of Deep Network Structure
      pythonmachine-learningdeep-learning
      pythonmachine-learningdeep-learningtensorflowautomlneural-architecture-search
      Python
      •
      Apache License 2.0
      •152•1k•11•12•Updated Oct 5, 2026Oct 5, 2026

    • zapbench

      Public
      The Zebrafish Activity Prediction Benchmark measures progress on the problem of predicting cellular-resolution neural activity throughout an entire vertebrate b…
      benchmarkmachine-learningtime-series
      benchmarkmachine-learningtime-seriesneuroscienceforecastingcalcium-imaginglight-sheet-microcopy
      Python
      •
      Apache License 2.0
      •21•80•3•6•Updated Oct 5, 2026Oct 5, 2026

    • connectomics

      Public
      connectomics
      connectomics
      Python
      •
      Apache License 2.0
      •17•66•5•20•Updated Oct 5, 2026Oct 5, 2026

    • weatherbench2

      Public
      A benchmark for the next generation of data-driven global weather models.
      Python
      •
      Apache License 2.0
      •79•639•51•46•Updated Oct 5, 2026Oct 5, 2026

    • e3x

      Public
      E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
      Python
      •
      Apache License 2.0
      •12•126•1•4•Updated Oct 5, 2026Oct 5, 2026

    • arco-era5

      Public
      Recipes for reproducing Analysis-Ready & Cloud Optimized (ARCO) ERA5 datasets.
      open-scienceecmwfzarr
      open-scienceecmwfzarrera5cloud-optimized
      Python
      •
      Apache License 2.0
      •43•508•21•2•Updated Oct 5, 2026Oct 5, 2026

    • android_world

      Public
      AndroidWorld is an environment and benchmark for autonomous agents
      Python
      •
      Apache License 2.0
      •193•945•41•12•Updated Oct 5, 2026Oct 5, 2026

    • text-to-text-transfer-transformer

      Public
      Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"
      Python
      •
      Apache License 2.0
      •799•6.6k•60•50•Updated Oct 5, 2026Oct 5, 2026

    • skai

      Public
      SKAI is a machine learning based tool for performing automatic building damage assessments on aerial imagery of disaster sites.
      Python
      •
      Apache License 2.0
      •24•166•2•28•Updated Oct 5, 2026Oct 5, 2026

    • veriharness

      Public
      Python
      •
      Apache License 2.0
      •6•38•1•1•Updated Oct 2, 2026Oct 2, 2026

    • mseb

      Public
      Python
      •
      Apache License 2.0
      •14•70•3•3•Updated Sep 30, 2026Sep 30, 2026

    • timesfm

      Public
      TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.
      Python
      •
      Apache License 2.0
      •3.3k•34k•159•107•Updated Sep 29, 2026Sep 29, 2026

    • visu3d

      Public
      3d without friction (Torch, TF, Jax, Numpy)
      plotlynerf3d
      plotlynerf3d
      Python
      •
      Apache License 2.0
      •15•170•3•12•Updated Oct 5, 2026Oct 5, 2026

    • ingestables

      Public
      Python
      •
      Apache License 2.0
      •4•13•1•2•Updated Sep 29, 2026Sep 29, 2026

    • dataclass_array

      Public
      Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
      tensorflownumpytorch
      tensorflownumpytorchjaxdataclassesdataclass
      Python
      •
      Apache License 2.0
      •4•54•1•5•Updated Oct 5, 2026Oct 5, 2026

    • optformer

      Public
      Python
      •
      Apache License 2.0
      •36•243•3•6•Updated Oct 5, 2026Oct 5, 2026

    • tf-slim

      Public
      Python
      •
      Apache License 2.0
      •99•375•12•1•Updated Sep 28, 2026Sep 28, 2026

    • scenic

      Public
      Scenic: A Jax Library for Computer Vision Research and Beyond
      researchcomputer-visiondeep-learning
      researchcomputer-visiondeep-learningtransformersattentionjaxvision-transformer
      Python
      •
      Apache License 2.0
      •483•3.8k•172•139•Updated Sep 28, 2026Sep 28, 2026

    • rrsi

      Public
      Python
      •
      Apache License 2.0
      •122•1.3k•5•6•Updated Sep 23, 2026Sep 23, 2026
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.