search

apax-hub / apax

A flexible and performant framework for training machine learning potentials.
https://apax.readthedocs.io
MIT License
18 stars 3 forks source link

incompatibility with jax 0.4.35 #360

Closed M-R-Schaefer closed 2 days ago

M-R-Schaefer commented 1 month ago

they removed experimental.host_callback.barrier_wait

M-R-Schaefer commented 2 days ago

closed by #371