Machine Learning Engineer
SweedFull Time
Mid-level (3 to 4 years)
Key technologies and capabilities for this role
Common questions about this position
The role requires 10+ years of full-time industry experience in large-scale MLOps and AI infrastructure.
Candidates need proven experience with distributed training systems using PyTorch, JAX, or TensorFlow, deep familiarity with reinforcement learning algorithms like PPO, SAC, or Q-learning, and strong programming skills with a track record of training deep learning models at scale.
A Bachelor's degree in Computer Science, Robotics, Engineering, or a related field, or equivalent experience is required.
NVIDIA has an open and nurturing atmosphere for research that encourages collaboration, with opportunities to publish at top venues and work with the broader scientific community.
This information is not specified in the job description.
Designs GPUs and AI computing solutions
NVIDIA designs and manufactures graphics processing units (GPUs) and system on a chip units (SoCs) for various markets, including gaming, professional visualization, data centers, and automotive. Their products include GPUs tailored for gaming and professional use, as well as platforms for artificial intelligence (AI) and high-performance computing (HPC) that cater to developers, data scientists, and IT administrators. NVIDIA generates revenue through the sale of hardware, software solutions, and cloud-based services, such as NVIDIA CloudXR and NGC, which enhance experiences in AI, machine learning, and computer vision. What sets NVIDIA apart from competitors is its strong focus on research and development, allowing it to maintain a leadership position in a competitive market. The company's goal is to drive innovation and provide advanced solutions that meet the needs of a diverse clientele, including gamers, researchers, and enterprises.