User input "gpu" is inplaced renamed to "cuda"

[PicoCentauri]

If one picks a device as gpu, multi-gpu we change this input inplace to "cuda", "mps" or"multi-cuda" as in

https://github.com/lab-cosmo/metatrain/blob/96f2c74f0a371845497604d7d594d1db6b273a1d/src/metatrain/utils/devices.py#L54-L58

and

https://github.com/lab-cosmo/metatrain/blob/96f2c74f0a371845497604d7d594d1db6b273a1d/src/metatrain/utils/devices.py#L68-L69

I think this is not ideal because probable later error raises will confront the user with a device "cuda" or "multi-cuda" which he did not specify in the input. Maybe it is better to store an internal name like resolved_device and not override the user input.

The linked code above will should be changed to

resolved_device = desired_device

if desired_device == "gpu":
    if torch.cuda.is_available():
        resolved_device = "cuda"
    elif _mps_is_available():
        resolved_device = "mps"

and

 if desired_device == "multi-gpu": 
     resolved_device = "multi-cuda"