tch
pub fn autocast<T, F>(enabled: bool, f: F) -> T where F: FnOnce() -> T,
Runs a closure in mixed precision.