bert
TensorFlow code and pre-trained models for BERTgoogle-research
Google Researchtuning_playbook
A playbook for systematically maximizing the performance of deep learning models.vision_transformer
text-to-text-transfer-transformer
Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"arxiv-latex-cleaner
arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXivsimclr
SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learnersmultinerf
A Code Release for Mip-NeRF 360, Ref-NeRF, and RawNeRFfootball
Check out the new game server:albert
ALBERT: A Lite BERT for Self-supervised Learning of Language Representationsscenic
Scenic: A Jax Library for Computer Vision Research and Beyondframe-interpolation
FILM: Frame Interpolation for Large Motion, In ECCV 2022.t5x
electra
ELECTRA: Pre-training Text Encoders as Discriminators Rather Than Generatorskubric
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.uda
Unsupervised Data Augmentation (UDA)pegasus
language
Shared repository for open-sourced projects from the Google AI Language team.dex-lang
Research language for array processing in the Haskell/ML familyparti
big_vision
Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.big_transfer
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.torchsde
Differentiable SDE solvers with GPU support and efficient sensitivity analysis.FLAN
disentanglement_lib
disentanglement_lib is an open-source library for research on learning disentangled representations.multilingual-t5
robotics_transformer
planet
Learning Latent Dynamics for Planning from Pixelsmixmatch
tapas
End-to-end neural table-text understanding models.fixmatch
A simple method to perform semi-supervised learning with limited data.morph-net
Fast & Simple Resource-Constrained Learning of Deep Network Structurededuplicate-text-datasets
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.batch-ppo
Efficient Batched Reinforcement Learning in TensorFlowaugmix
AugMix: A Simple Data Processing Method to Improve Robustness and Uncertaintymaxim
[CVPR 2022 Oral] Official repository for "MAXIM: Multi-Axis MLP for Image Processing". SOTA for denoising, deblurring, deraining, dehazing, and enhancement.magvit
Official JAX implementation of MAGVIT: Masked Generative Video Transformerpix2seq
Pix2Seq codebase: multi-tasks with generative modeling (autoregressive and diffusion)seed_rl
SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's architecture.meta-dataset
A dataset of datasets for learning to learn from few examplesnoisystudent
Code for Noisy Student Training. https://arxiv.org/abs/1911.04252jax3d
recsim
A Configurable Recommender Systems Simulation Platformlottery-ticket-hypothesis
A reimplementation of "The Lottery Ticket Hypothesis" (Frankle and Carbin) on MNIST.rliable
[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.circuit_training
long-range-arena
Long Range Arena for Benchmarking Efficient Transformersfederated
A collection of Google research projects related to Federated Learning and Federated Analytics.nasbench
NASBench: A Neural Architecture Search Dataset and Benchmarkprompt-tuning
Original Implementation of Prompt Tuning from Lester, et al, 2021bleurt
BLEURT is a metric for Natural Language Generation based on transfer learning.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.lasertagger
sound-separation
dreamer
Dream to Control: Learning Behaviors by Latent Imaginationrobopianist
[CoRL '23] Dexterous piano playing with deep reinforcement learning.pix2struct
fast-soft-sort
Fast Differentiable Sorting and Rankingbigbird
Transformers for Longer Sequencesravens
Train robotic agents to learn pick and place with deep learning for vision-based manipulation in PyBullet. Transporter Nets, CoRL 2020.sam
vmoe
batch_rl
Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 gamestensor2robot
Distributed machine learning infrastructure for large-scale robotics researchmint
Multi-modal Content Creation Model Training Infrastructure including the FACT model (AI Choreographer) implementation.byt5
adapter-bert
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.robustness_metrics
maxvit
[ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmentation, image quality, and generative modeling...receptive_field
Compute receptive fields of your favorite convnetsssl_detection
Semi-supervised learning for object detectionmaskgit
Official Jax Implementation of MaskGITl2p
Learning to Prompt (L2P) for Continual Learning @ CVPR22 and DualPrompt: Complementary Prompting for Rehearsal-free Continual Learning @ ECCV22nerf-from-image
Shape, Pose, and Appearance from a Single Image via Bootstrapped Radiance Field Inversioncomputation-thru-dynamics
Understanding computation in artificial and biological recurrent networks through the lens of dynamical systems.tf-slim
realworldrl_suite
Real-World RL Benchmark Suitedistilling-step-by-step
rigl
End-to-end training of sparse deep neural networks with little-to-no performance loss.python-graphs
A static analysis library for computing graph representations of Python programs suitable for use with graph neural networks.weatherbench2
A benchmark for the next generation of data-driven global weather models.tensorflow_constrained_optimization
task_adaptation
exoplanet-ml
Machine learning models and utilities for exoplanet science.ibc
Official implementation of Implicit Behavioral Cloning, as described in our CoRL 2021 paper, see more at https://implicitbc.github.io/self-organising-systems
tensorflow-coder
vdm
retvec
RETVec is an efficient, multilingual, and adversarially-robust text vectorizer.sparf
This is the official code release for SPARF: Neural Radiance Fields from Sparse and Noisy Poses [CVPR 2023-Highlight]falken
Falken provides developers with a service that allows them to train AI that can play their gamessyn-rep-learn
Learning from synthetic data - code and modelslm-extraction-benchmark
meliad
3d-moments
Code for CVPR 2022 paper '3D Moments from Near-Duplicate Photos'perceiver-ar
ott
language-table
Suite of human-collected datasets and a multi-task continuous control benchmark for open vocabulary visuolinguomotor learning.Love Open Source and this site? Check out how you can help us