Nvidia is hiring a
Senior AI Platform Engineer
At NVIDIA, we pride ourselves on data-driven decision-making, and the data science platform team is at the heart of this initiative! We are looking for an excellent Principal ML/AI Engineer to enhance our AI platform which process 10+ Trillion of events per month. This role will work closely with data science, data engineering, and product teams to ensure delivery of reliable services. Excellent opportunity to architect state-of-the-art tech, including machine learning at scale, containerized environments and GPU based inferencing. The candidate will have the opportunity to optimize and expand our data science platform and organization for model generation, orchestration, deployment, health, operations and metrics.
What you’ll be doing:
Design, architect MLOps lifecycle for our data science platform
Own production machine learning pipelines both in-house and cloud
Design, architect our GPU inferencing service both offline and online
Implement machine learning deployment tools, observability and APIs to build self-serve ML pipelines
Work closely with data scientists and data engineers to advise on implementation and provide trainings
Work with internal and external product teams to enable new features for GPU based inferencing
What we need to see:
Bachelors or Master’s degree in Computer Science or a related technical field or equivalent experience
10+ years of software engineering experience
Proficient understanding of distributed computing principles
Excellent understanding of ML Ops best practices, ability to design, measure, track, validate models and orchestrate
Expert level SW development skills in one or more: Java/Scala/Python/Go
Strong background with deploying ML/DL models at scale and inferencing stacks
Strong background with state-of-the-art neural network architectures (CNN, RNN, GAN, LLMs etc.) and experience in developing or using major deep learning frameworks (e.g. TensorFlow, PyTorch etc)
Experience designing and implementing low latency, high throughput applications on K8s
Excellent communication skills including the ability to identify and communicate data driven insights
Ways to stand out from the crowd:
Knowledge on building k8s custom operators
Experience with SLURM
Contributions to open source
NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you!The base salary range is $216,000 - $414,000. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
You will also be eligible for equity and benefits.
Please mention that you found the job on ARVR OK. Thanks.