Remove the deprecated import jax.tree_map#1775

Merged

OlaRonning merged 2 commits intopyro-ppl:masterfrom fehiepsi:tree1

Apr 6, 2024

+8-6

Commits on Apr 4, 2024

remove deprecated import jax.tree_map
fehiepsi
committed

Commits on Apr 6, 2024

fix enable_x64 logic
fehiepsi
committed