pub type LoadFn<'a> = &'a dyn Fn(&str) -> Result<Vec<f32>>;
A tensor as the checkpoint holds it: full (unsharded) BF16 values on the host.