pub fn sgd_state_from_map(map: &HashMap<String, Tensor>) -> HashMap<u64, Tensor>
Restore SGD velocity buffers from a string-keyed map.