Files
deepmind-research/physics_inspired_models/requirements.txt
Alex Botev 2c7c401024 Open sourcing the physics inspired models code.
PiperOrigin-RevId: 408640606
2021-11-09 22:56:08 +00:00

11 lines
236 B
Plaintext

git+https://github.com/deepmind/dm_hamiltonian_dynamics_suite@main#egg=dm_hamiltonian_dynamics_suite
absl-py==0.12.0
numpy>=1.16.4
scikit-learn>=1.0
typing>=3.7.4.3
jax==0.2.20
jaxline==0.0.3
distrax==0.0.2
optax==0.0.6
dm-haiku==0.0.3