Skip to content

Remove workaround for flushing jax queue

Vicentini Filippo requested to merge PhilipVinc/bump-mpi4jax into v3.0

Created by: PhilipVinc

Following mpi4jax#25 mpi4jax automatically flushes jax queues at exit, so we don't need the workaround anymore in netket itself . The same mechanism will be used to flush GPU operation queues in the future.

Merge request reports