neural_dynamics.config

Device selection utilities for the neural_dynamics package.

DEVICE: Final[torch.device] = device(type='cpu')

def get_device() -> torch.device:

Return the preferred torch device (MPS, CUDA, or CPU).