⚠ Enable if model requires custom code execution

Device

Compute device for training