AttributeError: module 'jax' has no attribute 'tree_multimap' - 29 11월 2023 댓글 쓰기 Problem AttributeError: module 'jax' has no attribute 'tree_multimap' Solution Use jax.tree_util.tree_map or jax.tree_map instead… JAXPythonTroubleshooting