Closed nestordemeure closed 1 year ago
Introduced code to recycle memory when doing a reshaping operation with change_detectors with JAX gpu data.
change_detectors
Previously one had to allocate a new JAX array to deal with the shape change.
Introduced code to recycle memory when doing a reshaping operation with
change_detectorswith JAX gpu data.Previously one had to allocate a new JAX array to deal with the shape change.