Discover google-research/self-organising-systems Open Source project

1

bert

TensorFlow code and pre-trained models for BERT

Python

37,769

2

google-research

Google Research

Jupyter Notebook

33,759

3

tuning_playbook

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

26,593

4

vision_transformer

Jupyter Notebook

10,251

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,099

6

arxiv-latex-cleaner

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

Python

5,233

7

simclr

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

Jupyter Notebook

3,937

8

multinerf

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

Python

3,612

9

timesfm

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

Python

3,576

10

scenic

Scenic: A Jax Library for Computer Vision Research and Beyond

Python

3,295

11

football

Check out the new game server:

Python

3,260

12

albert

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

Python

3,209

13

frame-interpolation

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

Python

2,818

14

t5x

Python

2,656

15

electra

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

Python

2,325

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,312

17

big_vision

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

Jupyter Notebook

2,219

18

uda

Unsupervised Data Augmentation (UDA)

Python

2,131

19

language

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

Python

1,605

20

pegasus

Python

1,600

21

dex-lang

Research language for array processing in the Haskell/ML family

Haskell

1,581

22

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

Python

1,548

23

parti

1,538

24

big_transfer

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

Python

1,504

25

FLAN

Python

1,460

26

robotics_transformer

Python

1,337

27

disentanglement_lib

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

Python

1,311

28

multilingual-t5

Python

1,197

29

circuit_training

Python

1,151

30

tapas

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

Python

1,143

31

planet

Learning Latent Dynamics for Planning from Pixels

Python

1,134

32

mixmatch

Python

1,130

33

deduplicate-text-datasets

Rust

1,104

34

fixmatch

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

Python

1,094

35

morph-net

Fast & Simple Resource-Constrained Learning of Deep Network Structure

Python

1,016

36

maxim

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

Python

996

37

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

38

batch-ppo

Efficient Batched Reinforcement Learning in TensorFlow

Python

963

39

augmix

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

Python

951

40

magvit

Official JAX implementation of MAGVIT: Masked Generative Video Transformer

Python

947

41

pix2seq

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

Jupyter Notebook

865

42

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

793

43

meta-dataset

A dataset of datasets for learning to learn from few examples

Jupyter Notebook

762

44

noisystudent

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

Python

751

45

rliable

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

Jupyter Notebook

747

46

recsim

A Configurable Recommender Systems Simulation Platform

Python

739

47

jax3d

Python

733

48

long-range-arena

Long Range Arena for Benchmarking Efficient Transformers

Python

719

49

lottery-ticket-hypothesis

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

Python

706

50

federated

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

Python

675

51

bleurt

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

Python

651

52

prompt-tuning

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

Python

642

53

nasbench

NASBench: A Neural Architecture Search Dataset and Benchmark

Python

641

54

neuralgcm

Hybrid ML + physics model of the Earth's atmosphere

Python

641

55

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

631

56

lasertagger

Python

606

57

sound-separation

Python

603

58

pix2struct

Python

587

59

vmoe

Jupyter Notebook

569

60

dreamer

Dream to Control: Learning Behaviors by Latent Imagination

Python

568

61

robopianist

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

Python

562

62

omniglue

Code release for CVPR'24 submission 'OmniGlue'

Python

561

63

fast-soft-sort

Fast Differentiable Sorting and Ranking

Python

561

64

ravens

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

Python

560

65

sam

Python

551

66

batch_rl

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

Python

521

67

bigbird

Transformers for Longer Sequences

Python

518

68

tensor2robot

Distributed machine learning infrastructure for large-scale robotics research

Python

483

69

byt5

Python

477

70

adapter-bert

Python

476

71

mint

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

Python

465

72

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

73

robustness_metrics

Jupyter Notebook

442

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

436

75

receptive_field

Compute receptive fields of your favorite convnets

Python

434

76

maskgit

Official Jax Implementation of MaskGIT

Jupyter Notebook

429

77

weatherbench2

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

Python

420

78

l2p

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

Python

408

79

distilling-step-by-step

Python

407

80

ssl_detection

Semi-supervised learning for object detection

Python

398

81

nerf-from-image

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

Python

377

82

computation-thru-dynamics

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

Jupyter Notebook

369

83

tf-slim

Python

368

84

realworldrl_suite

Real-World RL Benchmark Suite

Python

341

85

python-graphs

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

Python

325

86

rigl

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

Python

314

87

task_adaptation

Python

310

88

ibc

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

Python

306

89

tensorflow_constrained_optimization

Python

300

90

syn-rep-learn

Learning from synthetic data - code and models

Python

294

91

arco-era5

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

Python

291

92

vdm

Jupyter Notebook

291

93

rlds

Jupyter Notebook

284

94

exoplanet-ml

Machine learning models and utilities for exoplanet science.

Python

283

95

retvec

RETVec is an efficient, multilingual, and adversarially-robust text vectorizer.

Jupyter Notebook

281

96

sparf

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

Python

279

97

tensorflow-coder

Python

275

98

lm-extraction-benchmark

Python

270

99

language-table

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

Jupyter Notebook

260

100

falken

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

Python

254

google-research/self-organising-systems

google-research

Reviews

Repository Details

Self-organising Systems

More Repositories