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 5+ years of experience, along with an MSc or PhD in CS, EE, or CSEE or equivalent experience.
Candidates need a strong background in Deep Learning, strong programming skills in Python and PyTorch, and experience with inference optimization techniques such as quantization and frameworks like TensorRT, TensorRT-LLM, vLLM, or SGLang.
This is a hybrid role.
Familiarity with deploying Deep Learning models in production using Docker or Triton Inference Server, CUDA programming experience, familiarity with diffusion models, and proven experience in analyzing, modeling, and tuning GPU workloads for inference and training will make you stand out.
You'll work closely with research scientists, software engineers, and hardware experts.
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.