Commit Graph

2 Commits

Author SHA1 Message Date
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
Luyu Wang 1243baa0b0 Release training scripts and samplers.
PiperOrigin-RevId: 390063136
2021-08-27 15:00:21 +01:00