Skip to content

sr_onthefly: switch dtype detection to jax.eval_shape

Vicentini Filippo requested to merge github/fork/inailuig/sr_onthefly_evalshape into master

Created by: inailuig

inspired by https://github.com/netket/netket/pull/539

Merge request reports