epraya.Jungarian#

Jungarian(cost)#

Wrap function to call the Scipy J-V method outside of JAX using ShapeDtypeStruct and pure_callback, in order to not interfere in the ADAM implementation.

Parameters:

  • cost (list) – Cost matrix of the eigenvectors and energies configuration

  • Returns

  • val (function) – Pure call back of the hungarian function.