Function adam_state_from_map 

pub fn adam_state_from_map(
    map: &HashMap<String, Tensor>,
) -> Vec<(u64, Tensor, Tensor, u64)>

Restore Adam/AdamW state from a string-keyed map.

Returns Vec<(param_id, first_moment, second_moment, step)>.