About this role
AI Research Engineer at Domyn SpA. Location: Milan or Italy. Role: training models, optimizing infrastructure, deploying models Requirements: PhD or equivalent experience; 5+ years AI research/engineering (or 2+ years with PhD). Expertise with PyTorch/TensorFlow/JAX, distributed training (PyTorch Distributed, Ray, DeepSpeed), CUDA, profiling tools, Python and C++/Rust, and production model deployment. Category: Research and Development (R&D) Seniority: Senior Level Tools: PyTorch, TensorFlow, JAX, PyTorch Distributed, Ray, DeepSpeed, CUDA, TensorBoard, Weights & Biases, NVIDIA Nsight, Python, C++, Rust, NVIDIA Blackwell GPUs Commitment: Full Time Workplace: Hybrid Languages: English