8e24fbbb29
Added new models.
...
PiperOrigin-RevId: 434780632
2022-05-26 17:43:12 +01:00
b2fa23c838
Allow alternative functions to load extra data.
...
PiperOrigin-RevId: 434495304
2022-05-26 17:43:03 +01:00
3cb13ea0b9
Use new MuJoCo python bindings to implement dm_control/mujoco.
...
PiperOrigin-RevId: 432958654
2022-05-26 17:42:53 +01:00
255e4e1256
Migrate away from using JaxTestCase in tests
...
Why? JaxTestCase is deprecated for use outside the JAX project as of version 0.3.1; see https://jax.readthedocs.io/en/latest/changelog.html#jax-0-3-1-feb-18-2022
PiperOrigin-RevId: 432390267
2022-05-26 17:42:43 +01:00
f30fae8dbc
Internal change
...
PiperOrigin-RevId: 431903167
2022-05-26 17:42:33 +01:00
ded41df440
Expose times_sampled in SampleInfo.
...
PiperOrigin-RevId: 430170716
2022-05-26 17:42:11 +01:00
1642ae3499
Release fusion_tcv
...
PiperOrigin-RevId: 429049702
2022-02-16 16:11:04 +00:00
96a3d4e682
Added a missing link to the pseudocode.
...
PiperOrigin-RevId: 429027217
2022-02-16 16:10:55 +00:00
7b17dd5dde
Add split_rng=False (current default) to HTM.
...
Haiku plans to make split_rng a required argument to hk.vmap in an upcoming
release. This change updates HTM to preserve the current behaviour. We
also handle the case where users are using a release of Haiku without the
split_rng option, for these users split_rng=False is implied.
PiperOrigin-RevId: 428454975
2022-02-16 16:10:43 +00:00
840bfe86a6
Silence upcoming type errors in Python 3.9.
...
PiperOrigin-RevId: 426515913
2022-02-16 16:10:33 +00:00
35be4a0bb9
JaxTestCase now sets jax_numpy_rank_promotion='raise' by default
...
PiperOrigin-RevId: 425896973
2022-02-16 16:10:24 +00:00
bc869b25d2
Enable the checkpoint path containing the DM21 weights to be manually specified.
...
PiperOrigin-RevId: 424835062
2022-02-16 16:10:14 +00:00
7b427b5161
Clarify the input sequence usage.
...
PiperOrigin-RevId: 424333047
2022-02-16 16:10:04 +00:00
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
766286a7a1
Add avae to README.md
...
PiperOrigin-RevId: 422739191
2022-02-16 16:09:44 +00:00
ca3f9e9599
Internal change
...
PiperOrigin-RevId: 419592537
2022-02-16 16:09:34 +00:00
41256fd5e9
Fix test failures caused by a NumPy 1.21 upgrade
...
PiperOrigin-RevId: 418208713
2022-02-16 16:09:25 +00:00
68473a2243
Update NFNet optim to reflect Haiku default of dict instead of FlatMapping.
...
PiperOrigin-RevId: 418005668
2022-02-16 16:09:15 +00:00
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
3f0d5ed1a0
Internal change
...
PiperOrigin-RevId: 417034217
2022-02-16 16:08:46 +00:00
fba48d1e44
Add methane example to the README to demonstrate typical benchmarking settings
...
PiperOrigin-RevId: 415617674
2021-12-10 23:16:57 +00:00
dd5cc9cbf7
Fix typo.
...
PiperOrigin-RevId: 415615189
2021-12-10 23:16:05 +00:00
3146ff9f5d
Fix dataset options usage.
...
PiperOrigin-RevId: 415295087
2021-12-10 23:15:11 +00:00
72c72d530f
Added density_functional_approximation_dm21 to deepmind-research repo
...
PiperOrigin-RevId: 415267993
2021-12-09 17:41:31 +00:00
d55816beb3
Formatting fixes and internal changes.
...
PiperOrigin-RevId: 414656804
2021-12-09 14:00:10 +00:00
826ff89f21
Update paper title on Github
...
PiperOrigin-RevId: 413933278
2021-12-03 16:04:56 +00:00
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
601f12ba12
Updating physics_inspired_models References with links.
...
PiperOrigin-RevId: 412418348
2021-12-03 16:02:43 +00:00
c6220061a7
Fix or ignore some pytype errors.
...
PiperOrigin-RevId: 412228893
2021-12-03 16:01:34 +00:00
2c7c401024
Open sourcing the physics inspired models code.
...
PiperOrigin-RevId: 408640606
2021-11-09 22:56:08 +00:00
9b751b7d20
Silence some pytype errors.
...
PiperOrigin-RevId: 408459881
2021-11-09 22:51:14 +00:00
86307bf254
Packaging BYOL directory so that it can be distributed with pip install.
...
PiperOrigin-RevId: 408361356
2021-11-09 22:51:14 +00:00
ab9e154b04
noisy label project change of paper name
...
PiperOrigin-RevId: 406782713
2021-11-09 22:51:14 +00:00
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
5f6b11299e
Added tandem_dqn to deepmind-research repo
...
PiperOrigin-RevId: 405653277
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).
...
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
3257aa3833
Added journal information to the citation info.
...
PiperOrigin-RevId: 399767956
2021-09-30 12:20:18 +01:00
88bf8624e1
Fix colab link for nowcasting notebook.
...
PiperOrigin-RevId: 399705337
2021-09-29 17:57:04 +01:00
bad66ddb95
Adding nowcasting to deepmind-research repo.
...
PiperOrigin-RevId: 399684696
2021-09-29 16:00:30 +01:00
4dcbcc2865
Updating Powerpropagation README
...
PiperOrigin-RevId: 399653061
2021-09-29 12:24:34 +01:00
ecae49a93f
Added powerpropagation to deepmind-research repo
...
PiperOrigin-RevId: 399207934
2021-09-29 11:55:56 +01:00
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
24ba20a43a
Release entries for the CAD sketch dataset added
...
PiperOrigin-RevId: 398235260
2021-09-23 13:12:38 +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.
PiperOrigin-RevId: 396784433
2021-09-17 17:53:07 +01:00
cff83be778
Adding counterfactual_fairness to deepmind-research repo
...
PiperOrigin-RevId: 396665335
2021-09-17 17:53:02 +01:00
f49eb487a7
release code for Encoders and Ensembles.
...
PiperOrigin-RevId: 396572932
2021-09-17 17:53:02 +01:00
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
96a13847c6
Update the paper link.
...
PiperOrigin-RevId: 395556906
2021-09-17 17:49:29 +01:00
08a26772ec
Update arXiv version + citation for OGB-LSC
...
PiperOrigin-RevId: 394217992
2021-09-17 17:49:29 +01:00
769bfdbeaf
Releasing code for "Towards mental time travel: a hierarchical..."
...
PiperOrigin-RevId: 393323193
2021-08-27 15:05:33 +01:00
Sven Gowal
Nimrod Gileadi
Jake VanderPlas
Piotr Stanczyk
Albin Cassirer
Diego de Las Casas
Karel Lenc
Tom Hennigan
Rebecca Chen
James Spencer
Ziga Avsec
Andrew Lampinen
Sumedh Ghaisas
Peter Hawkins
Loren Maggiore
Alexander Gaunt
Adam Liska
Aleksandar Botev
DeepMind Team
Yazhe Li
Dong Yin
Tom Ward
Georg Ostrovski
Suman Ravuri
Jonathan Schwarz
Arthur Guez
Yaroslav Ganin
Ira Ktena
Benjamin Beyret
Luyu Wang
Petar Veličković