deepmind-research

KO | EN

gitlite — search

#deep-learning #video #linux #animation #chatgpt #terminal #css #hacktoberfest #docker #ai

deepmind-research

(★ 15,157)

This repository contains implementations and illustrative code to accompany DeepMind publications

File Explorer

ZIP Download

- - README.md
- - README.md
- - attacks.py
  - datasets.py
  - eval.py
  - experiment.py
  - experiment_test.py
  - model_zoo.py
  - train.py
  - utils.py
- - eval.py
  - model_zoo.py
  - README.md
- README.md
- requirements.txt
- run.sh
- AffordancesInContinuousEnvironment.ipynb
- AffordancesInDiscreteEnvironment.ipynb
- README.md
- requirements.txt
- asa_output.py
- config_dict.py
- contacts.py
- contacts_dataset.py
- contacts_experiment.py
- contacts_network.py
- distogram_io.py
- ensemble_contact_maps.py
- parsers.py
- paste_contact_maps.py
- README.md
- requirements.txt
- run_eval.sh
- secstruct.py
- test_domains.txt
- train_domains.txt
- two_dim_convnet.py
- two_dim_resnet.py
- avae_image.png
- checkpointer.py
- data_iterators.py
- decoders.py
- encoders.py
- kl.py
- README.md
- requirements.txt
- run.sh
- train.py
- train_main.py
- types.py
- vae.py
- README.md
- __init__.py
- dmlab_assets.py
- explore.py
- predicate_task.py
- predicate_task_test.py
- predicates.py
- README.md
- setup.py
- task_examples.py
- - __init__.py
  - byol.py
  - eval.py
- - __init__.py
  - augmentations.py
  - checkpointing.py
  - dataset.py
  - helpers.py
  - networks.py
  - optimizers.py
  - schedules.py
- byol_experiment.py
- eval_experiment.py
- main_loop.py
- main_loop_test.py
- README.md
- requirements.txt
- setup.py
- - sketch_data.gif
- common.proto
- constraints.proto
- download_dataset.sh
- entities.proto
- example.proto
- README.md
- __init__.py
- arm_opener.py
- ball_toss.py
- explore.py
- mocap_data.h5
- mocap_data.py
- props.py
- README.md
- setup.py
- task_examples.py
- tasks.png
- trajectories.py
- warehouse.py
- Causal_Reasoning_in_Probability_Trees.ipynb
- README.md
- CMTouch_Dataset_Visulization.ipynb
- download_datasets.sh
- README.md
- encoders_and_ensembles.ipynb
- README.md
- adult.py
- adult_pscf.py
- adult_pscf_config.py
- causal_network.py
- download_dataset.sh
- README.md
- requirements.txt
- run_adult_pscf.sh
- utils.py
- variational.py
- - gan_test.py
- cs.py
- file_utils.py
- gan.py
- image_metrics.py
- main.py
- main_cs.py
- main_ode.py
- nets.py
- README.md
- requirements.txt
- run.sh
- run_ode.sh
- utils.py
- layers.py
- model.py
- README.md
- requirements.txt
- train_main.py
- train_sup.py
- train_unsup.py
- training.py
- unit_test.py
- utils.py
- - dm21_aot_compiled_example.cc
  - dm21_aot_compiled_example.h
  - run_dm21_aot_compiled_example.cc
- - __init__.py
  - compute_hfx_density.py
  - compute_hfx_density_test.py
  - export_saved_model.py
  - neural_numint.py
  - neural_numint_test.py
- - tf_bazel.patch
- .bazelrc
- .bazelversion
- BUILD.bazel
- README.md
- requirements.txt
- requirements_aot_compilation.txt
- run.sh
- setup.py
- WORKSPACE.bazel
- attention_module.py
- enformer-training.ipynb
- enformer-usage.ipynb
- enformer.py
- enformer_test.py
- README.md
- requirements.txt
- cifar10_medium_cnn_experiments.ipynb
- README.md
- frcl.ipynb
- README.md
- agent.py
- combiners.py
- combiners_test.py
- environment.py
- experiments.py
- experiments_test.py
- fge_octave.py
- fge_state.py
- named_array.py
- named_array_test.py
- noise.py
- param_variation.py
- README.md
- ref_gen.py
- references.py
- references_main.py
- requirements.txt
- rewards.py
- rewards_used.py
- run_loop.py
- shape.py
- shapes_known.py
- targets.py
- tcv_common.py
- terminations.py
- trajectory.py
- transforms.py
- transforms_test.py
- antennae_helpers.py
- config.py
- evaluator.py
- helpers.py
- interpretability_helpers.py
- losses.py
- main.py
- model.py
- preprocessing.py
- README.md
- requirements.txt
- - dendritic_gated_network.ipynb
  - README.md
- - bernoulli_mnist.py
  - utils.py
  - utils_test.py
- base.py
- bernoulli.py
- bernoulli_test.py
- gaussian.py
- gaussian_test.py
- README.md
- requirements.txt
- run.sh
- data_writer.py
- geomancer.py
- geomancer_test.py
- README.md
- run.sh
- s3o4d.png
- setup.py
- train.py
- - t044_s09.ckpt.data-00000-of-00001
  - t044_s09.ckpt.index
  - t044_s09.ckpt.meta
- - test_large.pickle
  - test_small.pickle
- apply_binary.py
- graph_model.py
- graph_model_test.py
- README.md
- requirements.txt
- train.py
- train_binary.py
- train_test.py
- train_using_jax.py
- graph_matching_networks.ipynb
- README.md
- requirements.txt
- - animation.gif
  - architecture.jpg
- geco_utils.py
- HPU_Net.ipynb
- model.py
- model_test.py
- README.md
- run.sh
- setup.py
- unet_utils.py
- - htm_attention.py
  - htm_attention_test.py
- - ballet_environment.py
  - ballet_environment_core.py
  - ballet_environment_test.py
- README.md
- requirements.txt
- himo_example.ipynb
- README.md
- - code_overview.svg
  - realworld_clevr.png
  - teaser.png
- - __init__.py
  - data.py
  - decoder.py
  - distributions.py
  - factor_eval.py
  - iodine.py
  - networks.py
  - plotting.py
  - refinement.py
  - utils.py
- - tetrominoes_mini.tfrecords
- configurations.py
- download_checkpoints.sh
- Eval.ipynb
- main.py
- README.md
- requirements.txt
- run.sh
- - common.py
  - graph_matcher_test.py
  - tracer_test.py
- __init__.py
- curvature_blocks.py
- distributions.py
- estimator.py
- example.py
- layers_and_loss_tags.py
- loss_functions.py
- optimizer.py
- README.md
- requirements.txt
- run.sh
- setup.py
- tag_graph_matcher.py
- tracer.py
- utils.py
- - figure_5a_work_values.dat
  - figure_5b_df_bar.dat
  - figure_5b_df_lbar.dat
  - figure_5b_df_mbar.dat
- lammps.dat
- README.md
- - water_ramps_rollout.gif
- connectivity_utils.py
- download_dataset.sh
- graph_network.py
- learned_simulator.py
- model_demo.py
- noise_utils.py
- reading_utils.py
- README.md
- render_rollout.py
- requirements.txt
- run.sh
- train.py
- load_memo_data.ipynb
- README.md
- cfd_eval.py
- cfd_model.py
- cloth_eval.py
- cloth_model.py
- common.py
- core_model.py
- dataset.py
- download_dataset.sh
- normalization.py
- plot_cfd.py
- plot_cloth.py
- README.md
- requirements.txt
- run.sh
- run_model.py
- - mmv_fig.png
- - mm_embeddings.py
  - normalization.py
  - resnet.py
  - s3d.py
  - s3d_test.py
  - tsm_resnet.py
  - tsm_resnet_test.py
  - tsm_utils.py
  - tsm_utils_test.py
  - types.py
- - checkpoint.py
  - ucf101_dataset.py
- config.py
- eval_ucf101.py
- README.md
- requirements.txt
- README.md
- agc_optax.py
- autoaugment.py
- base.py
- dataset.py
- experiment.py
- experiment_nf_regnets.py
- experiment_nfnets.py
- fixup_resnet.py
- nf_regnet.py
- nf_resnet.py
- nfnet.py
- nfnet_demo_colab.ipynb
- optim.py
- README.md
- requirements.txt
- resnet.py
- run.sh
- skipinit_resnet.py
- test.py
- utils.py
- noisy_label_datasets_and_rater_features.ipynb
- paradigm.png
- README.md
- Open_sourced_dataset_and_model_snapshot_for_precipitation_nowcasting.ipynb
- README.md
- model.py
- README.md
- requirements.txt
- run_model.py
- transformer.py
- odegan_mog16.ipynb
- README.md
- - batching_utils.py
  - config.py
  - csr_builder.py
  - data_utils.py
  - datasets.py
  - download_mag.py
  - ensemble_predictions.py
  - experiment.py
  - generate_validation_splits.py
  - losses.py
  - models.py
  - neighbor_builder.py
  - organize_data.sh
  - pca_builder.py
  - README.md
  - requirements.txt
  - run_preprocessing.sh
  - run_pretrained_eval.sh
  - run_training.sh
  - schedules.py
  - split_and_save_indices.py
  - sub_sampler.py
- - batching_utils.py
  - config.py
  - conformer_utils.py
  - dataset_utils.py
  - datasets.py
  - download_pcq.py
  - ensemble_predictions.py
  - experiment.py
  - generate_conformer_features.py
  - generate_validation_splits.py
  - model.py
  - README.md
  - requirements.txt
  - run_preprocessing.sh
  - run_pretrained_eval.sh
  - run_training.sh
- README.md
- - eval_keyboard_fig5.py
  - generate_figures.ipynb
  - regressed_agent.py
  - run_dqn_fig4b.py
  - run_dqn_fig5.py
  - run_regressed_w_fig4b.py
  - run_regressed_w_fig4c.py
  - run_regressed_w_with_phi_fig4c.py
  - run_true_w_fig4.py
  - run_true_w_fig6.py
  - train_keyboard.py
  - train_keyboard_with_phi.py
  - train_phi_model.py
- auto_reset_environment.py
- configs.py
- dqn_agent.py
- environment_wrappers.py
- experiment.py
- keyboard_agent.py
- keyboard_utils.py
- README.md
- requirements.txt
- run.sh
- run_dqn.py
- run_dqn_test.py
- run_ok.py
- run_ok_test.py
- scavenger.py
- smart_module.py
- - imagenet_classification.ipynb
  - masked_language_modelling.ipynb
  - optical_flow.ipynb
  - video_autoencoding.ipynb
- - autoaugment.py
  - dataset.py
  - experiment.py
  - launch_local.sh
  - utils.py
- bytes_tokenizer.py
- io_processors.py
- io_processors_test.py
- perceiver.py
- position_encoding.py
- README.md
- requirements.txt
- - __init__.py
  - autoregressive.py
  - base.py
  - common.py
  - deterministic_vae.py
  - dynamics.py
  - networks.py
- __init__.py
- eval_metric.py
- integrators.py
- jaxline_configs.py
- jaxline_train.py
- launch_all.sh
- launch_local.sh
- metrics.py
- README.md
- requirements.txt
- setup.py
- utils.py
- - __init__.py
  - board_games_test.py
  - go.py
  - go_logic.py
  - go_logic_test.py
  - jaco_arm_board_game.py
  - logic_base.py
  - tic_tac_toe.py
  - tic_tac_toe_logic.py
  - tic_tac_toe_logic_test.py
- - __init__.py
  - boxoban.py
  - mujoban.py
  - mujoban_level.py
  - mujoban_level_test.py
  - mujoban_pad.py
  - mujoban_test.py
  - props.py
- explore.py
- README.md
- requirements.txt
- README.md
- - example_samples.png
- - cone.obj
  - cube.obj
  - cylinder.obj
  - icosphere.obj
- data_utils.py
- model_test.py
- modules.py
- README.md
- run.sh
- sample-pretrained.ipynb
- setup.py
- training.ipynb
- powerpropagation.ipynb
- README.md
- mini_3task_col_patts_hexos.npz
- mini_3task_col_patts_pentos.npz
- mini_3task_col_patts_stripes.npz
- mini_between_hexos.npz
- mini_between_pentos.npz
- mini_between_stripes.npz
- PrediNet.ipynb
- README.md
- - example_mpg.png
  - example_osl.png
- memory_planning_game.py
- one_shot_streetlearn.py
- README.md
- requirements.txt
- README.md
- - tasks.png
- atari.py
- atari_dqn.ipynb
- atari_example.py
- bsuite.ipynb
- dm_control_suite.py
- dm_control_suite_crr.ipynb
- dm_control_suite_d4pg.ipynb
- dm_control_suite_example.py
- dmlab_r2d2.ipynb
- networks.py
- README.md
- requirements.txt
- rwrl.py
- rwrl_d4pg.ipynb
- rwrl_example.py
- - binary30.p
  - socrates.p
- - satore.scrbl
- - Clause.rkt
  - clause.rkt
  - confluence.rkt
  - interact.rkt
  - misc.rkt
  - rewrite-tree.rkt
  - saturation.rkt
  - stress-test1.rkt
  - trie.rkt
  - unification-tree.rkt
  - unification.rkt
- clause-format.rkt
- Clause.rkt
- clause.rkt
- info.rkt
- interact.rkt
- json-output.rkt
- log.rkt
- main.rkt
- misc.rkt
- README.md
- rewrite-tree.rkt
- saturation.rkt
- tptp.rkt
- trie.rkt
- unification-tree.rkt
- unification.rkt
- __init__.py
- discriminator_nets.py
- eval_metrics.py
- experiment.py
- generators.py
- losses.py
- reader.py
- README.md
- requirements.txt
- run.sh
- utils.py
- __init__.py
- agent.py
- agent_with_penalties.py
- file_loading.py
- plot_results.ipynb
- README.md
- requirements.txt
- results_summary.py
- run_experiment.py
- side_effects_penalty.py
- side_effects_penalty_test.py
- training.py
- __init__.py
- dataset_example.py
- download.sh
- example_data.tfrecords
- extract.sh
- metadata_schema.py
- README.md
- requirements.txt
- reward_example.py
- run.sh
- sketchy.py
- README.md
- requirements.txt
- synthetic_returns.py
- agent.py
- atari_data.py
- gym_atari.py
- losses.py
- losses_test.py
- networks.py
- parts.py
- processors.py
- README.md
- replay.py
- requirements.txt
- run.sh
- run_tandem.py
- README.md
- requirements.txt
- run.sh
- transporter.py
- transporter_example.ipynb
- transporter_test.py
- - active_visual_match.lua
  - image_utils.lua
  - key_to_door.lua
  - key_to_door_bluekey.lua
  - key_to_door_factory.lua
  - key_to_door_to_match.lua
  - latent_information_acquisition.lua
  - latent_information_acquisition_factory.lua
  - passive_visual_match.lua
  - README.md
  - two_keys_to_choose_factory.lua
  - two_negative_keys.lua
  - visual_match_factory.lua
- - avm_notvt.png
  - avm_notvt_gamma1.png
  - avm_tvt.png
  - ktd_notvt.png
  - ktd_notvt_gamma1.png
  - ktd_tvt.png
- - active_visual_match.py
  - common.py
  - env.py
  - game.py
  - human_player.py
  - key_to_door.py
  - objects.py
  - README.md
- batch_env.py
- losses.py
- main.py
- memory.py
- nest_utils.py
- README.md
- requirements.txt
- rma.py
- run.sh
- tvt_rewards.py
- install_dependencies.sh
- main.py
- README.md
- requirements.txt
- run.sh
- quick_eval_cifar.py
- README.md
- requirements.txt
- run.sh
- save_example_images.py
- tiny_200K_idxs.txt
- README.md
- VISR_ICLR2020.ipynb
- - build_vocab.py
  - compute_blue_score.py
  - download.sh
  - freebase_preprocess.py
  - visualize_graph.py
- main.py
- README.md
- requirements.txt
- setup.py
- updaters.py
- utils.py
.travis.yml
__init__.py
CONTRIBUTING.md
LICENSE
README.md

# Use via CDN

jsDelivr

jsDelivr serves any public GitHub repository as a CDN with zero setup. Pick a version and a file to get a ready-to-paste link and snippet.

Version

File

Link

Example

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

This repository contains implementations and illustrative code to accompany
DeepMind publications. Along with publishing papers to accompany research
conducted at DeepMind, we release open-source
[environments](https://deepmind.com/research/open-source/open-source-environments/),
[data sets](https://deepmind.com/research/open-source/open-source-datasets/),
and [code](https://deepmind.com/research/open-source/open-source-code/) to
enable the broader research community to engage with our work and build upon it,
with the ultimate goal of accelerating scientific progress to benefit society.
For example, you can build on our implementations of the
[Deep Q-Network](https://github.com/deepmind/dqn) or
[Differential Neural Computer](https://github.com/deepmind/dnc), or experiment
in the same environments we use for our research, such as
[DeepMind Lab](https://github.com/deepmind/lab) or
[StarCraft II](https://github.com/deepmind/pysc2).

If you enjoy building tools, environments, software libraries, and other
infrastructure of the kind listed below, you can view open positions to work in
related areas on our [careers page](https://deepmind.com/careers/).

For a full list of our publications, please see
https://deepmind.com/research/publications/

## Projects

*   [Magnetic control of tokamak plasmas through deep reinforcement learning](fusion_tcv), Nature 2022
*   [Pushing the Frontiers of Density Functionals by Solving the Fractional Electron Problem](density_functional_approximation_dm21), Science 2021
*   [Mind the Gap: Assessing Temporal Generalization in Neural Language Models](pitfalls_static_language_models), NeurIPS 2021
*   [The Difficulty of Passive Learning in Deep Reinforcement Learning](tandem_dqn), NeurIPS 2021
*   [Skilful precipitation nowcasting using deep generative models of radar](nowcasting), Nature 2021
*   [Compute-Aided Design as Language](cadl)
*   [Encoders and ensembles for continual learning](continual_learning)
*   [Towards mental time travel: a hierarchical memory for reinforcement learning agents](hierarchical_transformer_memory)
*   [Perceiver IO: A General Architecture for Structured Inputs & Outputs](perceiver)
*   [Solving Mixed Integer Programs Using Neural Networks](neural_mip_solving)
*   [A Realistic Simulation Framework for Learning with Label Noise](noisy_label)
*   [Rapid Task-Solving in Novel Environments](rapid_task_solving), ICLR 2021
*   [WikiGraphs: A Wikipedia - Knowledge Graph Paired Dataset](wikigraphs), TextGraphs 2021
*   [Behavior Priors for Efficient Reinforcement Learning](box_arrangement)
*   [Learning Mesh-Based Simulation with Graph Networks](meshgraphnets), ICLR 2021
*   [Open Graph Benchmark - Large-Scale Challenge (OGB-LSC)](ogb_lsc)
*   [Synthetic Returns for Long-Term Credit Assignment](synthetic_returns)
*   [A Deep Learning Approach for Characterizing Major Galaxy Mergers](galaxy_mergers)
*   [Better, Faster Fermionic Neural Networks](kfac_ferminet_alpha) (KFAC implementation)
*   [Object-based attention for spatio-temporal reasoning](object_attention_for_reasoning)
*   [Effective gene expression prediction from sequence by integrating long-range interactions](enformer)
*   [Satore: First-order logic saturation with atom rewriting](satore)
*   [Characterizing signal propagation to close the performance gap in unnormalized ResNets](nfnets), ICLR 2021
*   [Uncovering the Limits of Adversarial Training against Norm-Bounded Adversarial Examples](adversarial_robustness)
*   [Learning rich touch representations through cross-modal self-supervision](cmtouch), CoRL 2020
*   [Functional Regularisation for Continual Learning](functional_regularisation_for_continual_learning), ICLR 2020
*   [The Autoencoding Variational Autoencoder](avae), NeurIPS 2020
*   [Self-Supervised MultiModal Versatile Networks](mmv), NeurIPS 2020
*   [ODE-GAN: Training GANs by Solving Ordinary Differential Equations](ode_gan), NeurIPS 2020
*   [Algorithms for Causal Reasoning in Probability Trees](causal_reasoning)
*   [Gated Linear Networks](gated_linear_networks), NeurIPS 2020
*   [Value-driven Hindsight Modelling](himo), NeurIPS 2020
*   [Targeted free energy estimation via learned mappings](learned_free_energy_estimation), Journal of Chemical Physics 2020
*   [Learning to Simulate Complex Physics with Graph Networks](learning_to_simulate), ICML 2020
*   [Physically Embedded Planning Problems](physics_planning_games)
*   [PolyGen: PolyGen: An Autoregressive Generative Model of 3D Meshes](polygen), ICML 2020
*   [Bootstrap Your Own Latent](byol)
*   [Catch & Carry: Reusable Neural Controllers for Vision-Guided Whole-Body Tasks](catch_carry), SIGGRAPH 2020
*   [MEMO: A Deep Network For Flexible Combination Of Episodic Memories](memo), ICLR 2020
*   [RL Unplugged: Benchmarks for Offline Reinforcement Learning](rl_unplugged)
*   [Disentangling by Subspace Diffusion (GEOMANCER)](geomancer), NeurIPS 2020
*   [What can I do here? A theory of affordances in reinforcement learning](affordances_theory), ICML 2020
*   [Scaling data-driven robotics with reward sketching and batch reinforcement learning](sketchy), RSS 2020
*   [Path-Specific Counterfactual Fairness](counterfactual_fairness), AAAI 2019
*   [The Option Keyboard: Combining Skills in Reinforcement Learning](option_keyboard), NeurIPS 2019
*   [VISR - Fast Task Inference with Variational Intrinsic Successor Features](visr), ICLR 2020
*   [Unveiling the predictive power of static structure in glassy systems](glassy_dynamics), Nature Physics 2020
*   [Multi-Object Representation Learning with Iterative Variational Inference (IODINE)](iodine)
*   [AlphaFold CASP13](alphafold_casp13), Nature 2020
*   [Unrestricted Adversarial Challenge](unrestricted_advx)
*   [Hierarchical Probabilistic U-Net (HPU-Net)](hierarchical_probabilistic_unet)
*   [Training Language GANs from Scratch](scratchgan), NeurIPS 2019
*   [Temporal Value Transport](tvt), Nature Communications 2019
*   [Continual Unsupervised Representation Learning (CURL)](curl), NeurIPS 2019
*   [Unsupervised Learning of Object Keypoints (Transporter)](transporter), NeurIPS 2019
*   [BigBiGAN](bigbigan), NeurIPS 2019
*   [Deep Compressed Sensing](cs_gan), ICML 2019
*   [Side Effects Penalties](side_effects_penalties)
*   [PrediNet Architecture and Relations Game Datasets](PrediNet)
*   [Unsupervised Adversarial Training](unsupervised_adversarial_training), NeurIPS 2019
*   [Graph Matching Networks for Learning the Similarity of Graph Structured
    Objects](graph_matching_networks), ICML 2019
*   [REGAL: Transfer Learning for Fast Optimization of Computation Graphs](regal)
*   [Deep Ensembles: A Loss Landscape Perspective](ensemble_loss_landscape)
*   [Powerpropagation](powerpropagation)
*   [Physics Inspired Models](physics_inspired_models)

## Disclaimer

*This is not an official Google product.*

deepmind-research

Owner Repos

File Explorer

# Use via CDN

Was this content helpful?

Recent Feedback

deepmind-research

Owner Repos

File Explorer

# Use via CDN

Was this content helpful?

Recent Feedback

Download README