Commit Graph

8 Commits

Author SHA1 Message Date
Peter Hawkins 82a347438f Fix pytype failures related to teaching pytype about NumPy scalar types.
PiperOrigin-RevId: 519205179
2023-06-02 18:03:40 +01:00
Peter Hawkins 6f0ddef7da Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.
PiperOrigin-RevId: 512349622
2023-06-02 18:02:30 +01:00
Peter Hawkins c051e6a51d Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.
PiperOrigin-RevId: 511294746
2023-06-02 18:02:16 +01:00
Yilei Yang 3af71dd9a7 Make this code compatible with Python 3.10.
PiperOrigin-RevId: 473457191
2022-10-12 10:31:41 +00:00
Jake VanderPlas 6fcb84268e Use jax.tree_util.tree_map in place of deprecated tree_multimap.
The latter is a simple alias of the former, so this change is a no-op.

PiperOrigin-RevId: 461229165
2022-07-24 17:53:28 +01:00
Rebecca Chen 840bfe86a6 Silence upcoming type errors in Python 3.9.
PiperOrigin-RevId: 426515913
2022-02-16 16:10:33 +00:00
Aleksandar Botev 601f12ba12 Updating physics_inspired_models References with links.
PiperOrigin-RevId: 412418348
2021-12-03 16:02:43 +00:00
Alex Botev 2c7c401024 Open sourcing the physics inspired models code.
PiperOrigin-RevId: 408640606
2021-11-09 22:56:08 +00:00