Machine Learning Engineer
SweedFull Time
Mid-level (3 to 4 years)
Key technologies and capabilities for this role
Common questions about this position
A BS in Computer Science or equivalent and 8+ years of experience in large-scale software or infrastructure systems are required, with 5+ years dedicated to ML platforms or MLOps.
Expert knowledge of distributed training frameworks like PyTorch, TensorFlow, JAX, and orchestration systems such as Kubernetes, Slurm, Kubeflow, Airflow, MLflow is required, along with strong Python programming and experience in Go, C++, or Rust.
This information is not specified in the job description.
This information is not specified in the job description.
Candidates stand out with practical experience supporting research teams on new GPU hardware, contributions to open-source MLOps projects, and proficiency in optimizing multi-node training on large GPU clusters.
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.