pub trait Load { // Required method fn load(&self, device: &Device, dtype: Option<DType>) -> Result<Tensor>; }