Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-05-12 06:37:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Commit Graph

  • affa162ac7 Return exc, vxc as float64 arrays. James Spencer 2022-10-11 10:13:38 +00:00
  • 789bc38a2e Add unused omega argument to NeuralNumint.eval_xc to match pyscf interface James Spencer 2022-10-11 10:11:56 +00:00
  • da0f2de14d Replace jnp.DeviceArray with the new public type jax.Array. Yash Katariya 2022-09-28 21:53:07 +00:00
  • 58fb45db7e Make this code compatible with Python 3.10. Yilei Yang 2022-09-12 09:56:26 +00:00
  • 3af71dd9a7 Make this code compatible with Python 3.10. Yilei Yang 2022-09-10 13:13:36 +00:00
  • 586a1c55de fix side effects baseline test flakiness due to corner case Victoria Krakovna 2022-07-26 20:51:08 +00:00
  • 6fcb84268e Use jax.tree_util.tree_map in place of deprecated tree_multimap. Jake VanderPlas 2022-07-15 21:14:58 +01:00
  • 956c4b5d9c Use jax.tree_util.tree_map in place of deprecated tree_multimap. Jake VanderPlas 2022-07-14 22:03:46 +01:00
  • 11c2ab53e8 Update enformer-training.ipynb to show how to restore from checkpoint. Kyle Taylor 2022-05-25 12:37:13 +01:00
  • 57456a001d Minor changes to match trained SOTA model. Kyle Taylor 2022-05-24 18:36:03 +01:00
  • d436681054 Remove dependency on kernel-pruner. Saran Tunyasuvunakool 2022-05-18 13:44:34 +01:00
  • a97d8b1807 Remove conda install instructions for DM21 James Spencer 2022-05-10 13:52:55 +01:00
  • f4916dadab Remove unused comments related to Python 2 compatibility. Yilei Yang 2022-04-21 08:29:03 +01:00
  • 94702704c8 Remove unused comments related to Python 2 compatibility. Yilei Yang 2022-04-13 08:31:38 +01:00
  • 5501e3237b Remove unused comments related to Python 2 compatibility. Yilei Yang 2022-04-13 08:31:08 +01:00
  • 81dedafa1d Copybara import of the project: Sharad Vikram 2022-04-12 06:04:39 +01:00
  • 34b2486902 Remove unused comments related to Python 2 compatibility. Yilei Yang 2022-04-11 17:06:49 +01:00
  • 152ac280f2 Remove unused comments related to Python 2 compatibility. Yilei Yang 2022-04-11 16:21:51 +01:00
  • 9a526bd35b Remove unused comments related to Python 2 compatibility. Yilei Yang 2022-04-11 15:10:56 +01:00
  • 92a307a920 Explicitly import estimator from tensorflow as a separate import instead of accessing it via tf.estimator and depend on the tensorflow estimator target. DeepMind Team 2022-03-24 10:07:15 +00:00
  • 464939ede1 Silence some pytype errors. Rebecca Chen 2022-03-18 11:57:29 +00:00
  • 8e24fbbb29 Added new models. Sven Gowal 2022-03-15 16:59:51 +00:00
  • b2fa23c838 Allow alternative functions to load extra data. Sven Gowal 2022-03-14 17:06:39 +00:00
  • 3cb13ea0b9 Use new MuJoCo python bindings to implement dm_control/mujoco. Nimrod Gileadi 2022-03-07 17:04:10 +00:00
  • 255e4e1256 Migrate away from using JaxTestCase in tests Jake VanderPlas 2022-03-04 09:44:03 +00:00
  • f30fae8dbc Internal change Piotr Stanczyk 2022-03-02 12:05:48 +00:00
  • ded41df440 Expose times_sampled in SampleInfo. Albin Cassirer 2022-02-22 11:14:37 +00:00
  • 1642ae3499 Release fusion_tcv Diego de Las Casas 2022-02-16 16:08:00 +00:00
  • 96a3d4e682 Added a missing link to the pseudocode. Karel Lenc 2022-02-16 14:05:26 +00:00
  • 7b17dd5dde Add split_rng=False (current default) to HTM. Tom Hennigan 2022-02-14 10:33:11 +00:00
  • 840bfe86a6 Silence upcoming type errors in Python 3.9. Rebecca Chen 2022-02-05 00:19:23 +00:00
  • 35be4a0bb9 JaxTestCase now sets jax_numpy_rank_promotion='raise' by default Jake VanderPlas 2022-02-02 16:57:53 +00:00
  • bc869b25d2 Enable the checkpoint path containing the DM21 weights to be manually specified. James Spencer 2022-01-28 12:07:39 +00:00
  • 7b427b5161 Clarify the input sequence usage. Ziga Avsec 2022-01-26 14:39:37 +00:00
  • cdd3bdb636 Switch from jax.vmap to hk.vmap: this does not change behavior at the moment, but aligns better with API recommendations. Andrew Lampinen 2022-01-24 13:29:24 +00:00
  • 766286a7a1 Add avae to README.md Sumedh Ghaisas 2022-01-19 07:57:04 +00:00
  • ca3f9e9599 Internal change Tom Hennigan 2022-01-04 15:45:11 +00:00
  • 41256fd5e9 Fix test failures caused by a NumPy 1.21 upgrade Peter Hawkins 2021-12-25 00:55:29 +00:00
  • 68473a2243 Update NFNet optim to reflect Haiku default of dict instead of FlatMapping. Loren Maggiore 2021-12-23 15:34:25 +00:00
  • 4b1583ae7a Freeze version of dm-control in requirements.txt files. Nimrod Gileadi 2021-12-20 10:05:47 +00:00
  • 3f0d5ed1a0 Internal change Nimrod Gileadi 2021-12-17 18:27:35 +00:00
  • fba48d1e44 Add methane example to the README to demonstrate typical benchmarking settings Alexander Gaunt 2021-12-10 22:56:43 +00:00
  • dd5cc9cbf7 Fix typo. James Spencer 2021-12-10 22:42:38 +00:00
  • 3146ff9f5d Fix dataset options usage. Loren Maggiore 2021-12-09 18:14:41 +00:00
  • 72c72d530f Added density_functional_approximation_dm21 to deepmind-research repo James Spencer 2021-12-08 12:01:54 +00:00
  • d55816beb3 Formatting fixes and internal changes. Loren Maggiore 2021-12-07 09:58:30 +00:00
  • 826ff89f21 Update paper title on Github Adam Liska 2021-12-03 15:42:49 +00:00
  • 03be2142f9 Remove usages of apply_rng=True from Haiku code. Tom Hennigan 2021-12-01 16:46:33 +00:00
  • 601f12ba12 Updating physics_inspired_models References with links. Aleksandar Botev 2021-11-26 11:40:46 +00:00
  • c6220061a7 Fix or ignore some pytype errors. DeepMind Team 2021-11-25 09:45:10 +00:00
  • 2c7c401024 Open sourcing the physics inspired models code. Alex Botev 2021-10-27 00:57:04 +01:00
  • 9b751b7d20 Silence some pytype errors. Rebecca Chen 2021-11-08 23:44:22 +00:00
  • 86307bf254 Packaging BYOL directory so that it can be distributed with pip install. Yazhe Li 2021-11-08 17:20:41 +00:00
  • ab9e154b04 noisy label project change of paper name Dong Yin 2021-11-01 10:00:35 +00:00
  • d25484af7f Enable loading and processing OBJ meshes from memory. Tom Ward 2021-10-28 14:04:54 +01:00
  • 5f6b11299e Added tandem_dqn to deepmind-research repo Georg Ostrovski 2021-10-26 16:30:00 +01:00
  • f6da52cb38 [JAX] Replace uses of deprecated jax.ops.index_update(x, idx, y) APIs with their up-to-date, more succinct equivalent x.at[idx].set(y). Peter Hawkins 2021-10-06 14:57:06 +01:00
  • 3257aa3833 Added journal information to the citation info. Suman Ravuri 2021-09-29 21:52:59 +01:00
  • 88bf8624e1 Fix colab link for nowcasting notebook. Diego de Las Casas 2021-09-29 17:32:04 +01:00
  • bad66ddb95 Adding nowcasting to deepmind-research repo. Suman Ravuri 2021-09-29 15:47:23 +01:00
  • 4dcbcc2865 Updating Powerpropagation README Jonathan Schwarz 2021-09-29 12:23:02 +01:00
  • ecae49a93f Added powerpropagation to deepmind-research repo Jonathan Schwarz 2021-09-27 17:22:32 +01:00
  • 243aff2fd6 Update version requirements for constrained_optidice. Update launchpad launch type flag. Minor bug fix. Arthur Guez 2021-09-24 16:04:35 +01:00
  • 24ba20a43a Release entries for the CAD sketch dataset added Yaroslav Ganin 2021-09-22 15:32:57 +01:00
  • 41e2d45ed8 Fixed issue in cutmix where split_batch_size would be undefined. Allow users to easily specify that they do not wish to use extra data. Sven Gowal 2021-09-15 10:01:03 +01:00
  • cff83be778 Adding counterfactual_fairness to deepmind-research repo Ira Ktena 2021-09-14 21:11:14 +01:00
  • f49eb487a7 release code for Encoders and Ensembles. Benjamin Beyret 2021-09-14 13:30:24 +01:00
  • a7d75013c9 Remove users of jax.api.* symbols, in preparation for removing the deprecated jax.api name. Peter Hawkins 2021-09-10 00:32:39 +01:00
  • 96a13847c6 Update the paper link. Luyu Wang 2021-09-08 22:19:56 +01:00
  • 08a26772ec Update arXiv version + citation for OGB-LSC Petar Veličković 2021-09-01 14:42:15 +01:00
  • 769bfdbeaf Releasing code for "Towards mental time travel: a hierarchical..." Andrew Lampinen 2021-08-27 12:26:01 +01:00
  • 27c2ca3956 (Enformer notebook) Make variant score list unique. Ziga Avsec 2021-08-23 10:44:42 +01:00
  • 3d9f601c69 Update PolyGen installation to do single pip install & clone into /tmp. Tom Ward 2021-08-17 19:57:46 +01:00
  • 17b5dbaf88 Update README.md. Luyu Wang 2021-08-13 15:19:23 +01:00
  • 1243baa0b0 Release training scripts and samplers. Luyu Wang 2021-08-11 07:34:25 +01:00
  • 5e35a17cf4 Internal changes. Diego de Las Casas 2021-08-10 18:30:14 +01:00
  • 30799687ed Merge pull request #262 from ttt733:patch-1 alimuldal 2021-08-10 17:57:53 +01:00
  • 0015af5805 [tf.data] Moving autotuning options from tf.data.Options.experimental_optimization.autotune* to a newly created tf.data.Options.autotune.* and removing support for tf.data.Options.experimental_optimization.autotune_buffers. Jiri Simsa 2021-08-08 16:10:01 +01:00
  • 67455ed367 Release graph2text transformer modules. Luyu Wang 2021-08-06 17:39:22 +01:00
  • ac7a19a9b2 [JAX] Relax test tolerances. Peter Hawkins 2021-08-05 23:30:48 +01:00
  • 32c1427c74 Remove conflicting version requirements Trevor Thackston 2021-08-06 01:12:17 -05:00
  • 3eddada361 Fix video_autoencoding colab reshape issue. David Ding 2021-08-04 16:53:14 +01:00
  • a5522d0784 Update paper title for ALOE. David Ding 2021-08-04 16:45:11 +01:00
  • d96ef0b6ee Download byte_tokenizers.py in Perceiver language modeling colab. David Ding 2021-08-04 16:26:25 +01:00
  • 8cc5c39662 Bug fix for stitching function in optical flow colab. David Ding 2021-08-03 16:50:57 +01:00
  • 21084c8489 Fix Perceiver IO author list, add correct ImageNet paper section. Drew Jaegle 2021-08-03 12:24:01 +01:00
  • 887a4fcda6 Add links to colabs. David Ding 2021-08-03 12:14:05 +01:00
  • df9b857b4e Add cell to install dependencies to run notebooks on cloud colab. David Ding 2021-08-02 18:27:24 +01:00
  • 48290008d1 Added pre-activation ResNet-18 checkpoints. Sven Gowal 2021-08-02 13:57:35 +01:00
  • 414683cc12 Add arXiv link for Perceiver IO. David Ding 2021-08-02 09:05:35 +01:00
  • 2fede222dd Release Perceiver David Ding 2021-07-30 17:30:51 +01:00
  • ab87c9cd72 Follow common naming convention for all notebooks. David Ding 2021-07-30 17:13:29 +01:00
  • 316297879f Minor wording changes David Ding 2021-07-30 17:01:11 +01:00
  • 768d530bf9 Add ImageNet classification colab. David Ding 2021-07-30 16:01:30 +01:00
  • 12971d9a42 Example training pipeline for perceiver. David Ding 2021-07-29 22:45:43 +01:00
  • 8c431d40ea Add the initial language colab for inference only Sebastian Borgeaud 2021-07-29 19:09:31 +01:00
  • 310620d045 Add optical flow visualization notebook. David Ding 2021-07-29 18:25:58 +01:00
  • 595d0727b2 Add jax implementation of extract_patches. David Ding 2021-07-29 18:05:07 +01:00
  • 198cf845de Move make_cross_attention_mask into the primitives section. Sebastian Borgeaud 2021-07-29 15:05:33 +01:00
  • 0e552473e5 Add BytesTokenizer and EmbeddingDecoder Both are required for the language colab (coming next) Sebastian Borgeaud 2021-07-29 14:58:44 +01:00