Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-05-20 11:22:09 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
316 Commits 77 Branches 0 Tags
35be4a0bb90c2056fc96989ea181c8b0ec066d60
Commit Graph

316 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Jake VanderPlas 35be4a0bb9 JaxTestCase now sets jax_numpy_rank_promotion='raise' by default 
PiperOrigin-RevId: 425896973
 2022-02-16 16:10:24 +00:00
James Spencer bc869b25d2 Enable the checkpoint path containing the DM21 weights to be manually specified. 
PiperOrigin-RevId: 424835062
 2022-02-16 16:10:14 +00:00
Ziga Avsec 7b427b5161 Clarify the input sequence usage. 
PiperOrigin-RevId: 424333047
 2022-02-16 16:10:04 +00:00
Andrew Lampinen cdd3bdb636 Switch from jax.vmap to hk.vmap: this does not change behavior at the moment, but aligns better with API recommendations. 
PiperOrigin-RevId: 423789562
 2022-02-16 16:09:54 +00:00
Sumedh Ghaisas 766286a7a1 Add avae to README.md 
PiperOrigin-RevId: 422739191
 2022-02-16 16:09:44 +00:00
Tom Hennigan ca3f9e9599 Internal change 
PiperOrigin-RevId: 419592537
 2022-02-16 16:09:34 +00:00
Peter Hawkins 41256fd5e9 Fix test failures caused by a NumPy 1.21 upgrade 
PiperOrigin-RevId: 418208713
 2022-02-16 16:09:25 +00:00
Loren Maggiore 68473a2243 Update NFNet optim to reflect Haiku default of dict instead of FlatMapping. 
PiperOrigin-RevId: 418005668
 2022-02-16 16:09:15 +00:00
Nimrod Gileadi 4b1583ae7a Freeze version of dm-control in requirements.txt files. 
A new version of dm_control is about to come out, which will depend on MuJoCo 2.1.1 (instead of 2.1.0), and we don't want to break existing users of dm_control until they've upgraded their MuJoCo installation.

PiperOrigin-RevId: 417360341
 2022-02-16 16:09:03 +00:00
Nimrod Gileadi 3f0d5ed1a0 Internal change 
PiperOrigin-RevId: 417034217
 2022-02-16 16:08:46 +00:00
Alexander Gaunt fba48d1e44 Add methane example to the README to demonstrate typical benchmarking settings 
PiperOrigin-RevId: 415617674
 2021-12-10 23:16:57 +00:00
James Spencer dd5cc9cbf7 Fix typo. 
PiperOrigin-RevId: 415615189
 2021-12-10 23:16:05 +00:00
Loren Maggiore 3146ff9f5d Fix dataset options usage. 
PiperOrigin-RevId: 415295087
 2021-12-10 23:15:11 +00:00
James Spencer 72c72d530f Added density_functional_approximation_dm21 to deepmind-research repo 
PiperOrigin-RevId: 415267993
 2021-12-09 17:41:31 +00:00
Loren Maggiore d55816beb3 Formatting fixes and internal changes. 
PiperOrigin-RevId: 414656804
 2021-12-09 14:00:10 +00:00
Adam Liska 826ff89f21 Update paper title on Github 
PiperOrigin-RevId: 413933278
 2021-12-03 16:04:56 +00:00
Tom Hennigan 03be2142f9 Remove usages of apply_rng=True from Haiku code. 
This change is a no-op, `apply_rng=True` is the default value and for the last
year True has been the only accepted value for apply_rng. We intend to remove
this argument from Haiku in an upcoming change.

PiperOrigin-RevId: 413426614
 2021-12-03 16:03:49 +00:00
Aleksandar Botev 601f12ba12 Updating physics_inspired_models References with links. 
PiperOrigin-RevId: 412418348
 2021-12-03 16:02:43 +00:00
DeepMind Team c6220061a7 Fix or ignore some pytype errors. 
PiperOrigin-RevId: 412228893
 2021-12-03 16:01:34 +00:00
Alex Botev 2c7c401024 Open sourcing the physics inspired models code. 
PiperOrigin-RevId: 408640606
 2021-11-09 22:56:08 +00:00
Rebecca Chen 9b751b7d20 Silence some pytype errors. 
PiperOrigin-RevId: 408459881
 2021-11-09 22:51:14 +00:00
Yazhe Li 86307bf254 Packaging BYOL directory so that it can be distributed with pip install. 
PiperOrigin-RevId: 408361356
 2021-11-09 22:51:14 +00:00
Dong Yin ab9e154b04 noisy label project change of paper name 
PiperOrigin-RevId: 406782713
 2021-11-09 22:51:14 +00:00
Tom Ward d25484af7f Enable loading and processing OBJ meshes from memory. 
Also remove dependency on modules from data_utils to make it more hermetic.

PiperOrigin-RevId: 406120785
 2021-11-09 22:51:14 +00:00
Georg Ostrovski 5f6b11299e Added tandem_dqn to deepmind-research repo 
PiperOrigin-RevId: 405653277
 2021-10-26 16:30:00 +01:00
Peter Hawkins 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). 
The JAX operators:
jax.ops.index_update(x, jax.ops.index[idx], y)
jax.ops.index_add(x, jax.ops.index[idx], y)
...

have long been deprecated in lieu of their more succinct counterparts:
x.at[idx].set(y)
x.at[idx].add(y)
...

This change updates users of the deprecated APIs to use the current APIs, in preparation for removing the deprecated forms from JAX.

The main subtlety is that if `x` is not a JAX array, we must cast it to one using `jnp.asarray(x)` before using the new form, since `.at[...]` is only defined on JAX arrays.

PiperOrigin-RevId: 401233414
 2021-10-26 18:22:43 +01:00
Suman Ravuri 3257aa3833 Added journal information to the citation info. 
PiperOrigin-RevId: 399767956
 2021-09-30 12:20:18 +01:00
Diego de Las Casas 88bf8624e1 Fix colab link for nowcasting notebook. 
PiperOrigin-RevId: 399705337
 2021-09-29 17:57:04 +01:00
Suman Ravuri bad66ddb95 Adding nowcasting to deepmind-research repo. 
PiperOrigin-RevId: 399684696
 2021-09-29 16:00:30 +01:00
Jonathan Schwarz 4dcbcc2865 Updating Powerpropagation README 
PiperOrigin-RevId: 399653061
 2021-09-29 12:24:34 +01:00
Jonathan Schwarz ecae49a93f Added powerpropagation to deepmind-research repo 
PiperOrigin-RevId: 399207934
 2021-09-29 11:55:56 +01:00
Arthur Guez 243aff2fd6 Update version requirements for constrained_optidice. 
Update launchpad launch type flag.
Minor bug fix.

PiperOrigin-RevId: 398732217
 2021-09-29 11:54:51 +01:00
Yaroslav Ganin 24ba20a43a Release entries for the CAD sketch dataset added 
PiperOrigin-RevId: 398235260
 2021-09-23 13:12:38 +01:00
Sven Gowal 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.

PiperOrigin-RevId: 396784433
 2021-09-17 17:53:07 +01:00
Ira Ktena cff83be778 Adding counterfactual_fairness to deepmind-research repo 
PiperOrigin-RevId: 396665335
 2021-09-17 17:53:02 +01:00
Benjamin Beyret f49eb487a7 release code for Encoders and Ensembles. 
PiperOrigin-RevId: 396572932
 2021-09-17 17:53:02 +01:00
Peter Hawkins a7d75013c9 Remove users of jax.api.* symbols, in preparation for removing the deprecated jax.api name. 
In most cases, use the public jax.* name instead.

PiperOrigin-RevId: 395812511
 2021-09-17 17:49:29 +01:00
Luyu Wang 96a13847c6 Update the paper link. 
PiperOrigin-RevId: 395556906
 2021-09-17 17:49:29 +01:00
Petar Veličković 08a26772ec Update arXiv version + citation for OGB-LSC 
PiperOrigin-RevId: 394217992
 2021-09-17 17:49:29 +01:00
Andrew Lampinen 769bfdbeaf Releasing code for "Towards mental time travel: a hierarchical..." 
PiperOrigin-RevId: 393323193
 2021-08-27 15:05:33 +01:00
Ziga Avsec 27c2ca3956 (Enformer notebook) Make variant score list unique. 
PiperOrigin-RevId: 392387922
 2021-08-27 15:03:30 +01:00
Tom Ward 3d9f601c69 Update PolyGen installation to do single pip install & clone into /tmp. 
This fixes issues with dep versioning resolution between each package.

PiperOrigin-RevId: 391343870
 2021-08-27 15:02:27 +01:00
Luyu Wang 17b5dbaf88 Update README.md. 
PiperOrigin-RevId: 390606818
 2021-08-27 15:01:24 +01:00
Luyu Wang 1243baa0b0 Release training scripts and samplers. 
PiperOrigin-RevId: 390063136
 2021-08-27 15:00:21 +01:00
Diego de Las Casas 5e35a17cf4 Internal changes. 
PiperOrigin-RevId: 389919323
 2021-08-27 14:59:16 +01:00
alimuldal 30799687ed Merge pull request #262 from ttt733:patch-1 
PiperOrigin-RevId: 389898784
 2021-08-10 17:57:53 +01:00
Jiri Simsa 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. 
PiperOrigin-RevId: 389487692
 2021-08-09 17:19:38 +01:00
Luyu Wang 67455ed367 Release graph2text transformer modules. 
PiperOrigin-RevId: 389194661
 2021-08-09 17:18:31 +01:00
Peter Hawkins ac7a19a9b2 [JAX] Relax test tolerances. 
These tests fail with an upcoming change to JAX that adds `jit` decorators around a number of JAX standard library functions. This appears to cause some very small numerical changes, most likely related to slightly different optimization opportunities available to the compiler.

PiperOrigin-RevId: 389039057
 2021-08-09 17:17:21 +01:00
Trevor Thackston 32c1427c74 Remove conflicting version requirements 2021-08-06 01:12:17 -05:00