Skip to main content

Senior Python Deep Learning Automation Engineer

Senior Python Deep Learning Automation Engineer
2100 NVIDIA USA

Senior Python DL Automation Engineer, Deep Learning Algorithms

Senior Python DL Automation Engineer, Deep Learning Algorithms
Apply locations US, CA, Santa Clara
US, CA, Remote
Time type Full time
Posted on Posted 3 Days Ago
Job requisition id JR(phone number removed)

We are now looking for a Senior Python Automation Engineer, for our Deep Learning Algorithms team! Join the team building software which will be used by the entire world of AI. Work with high class software engineers to implement a large scale toolset that tests deep learning models and frameworks on the most powerful computers. Ability to work in a multifaceted, fast-paced environment is required as well as strong social skills. In this role you will be interacting with internal partners, users, and members of the open source community to implement solutions for building, testing, integrating, and releasing of NVIDIA AI Services and Deep Learning Frameworks on the most powerful, enterprise-grade GPU clusters capable of hundreds of Peta FLOPS. This role spans multiple products such as PyTorch, TensorFlow, JAX, PaddlePaddle. You will work with internal engineering teams to deploy and operationalize AI models and services at scale by driving adoption for end-to-end Machine Learning and Deep Learning solutions in the cloud and on prem.

What you'll be doing:

  • Automating and optimizing testing of Deep Learning models and AI Services from different data domains with focus on inference
  • Developing shared utilities for setting up systems, running tests, recording results and visualization on dashboards.
  • Configuring, maintaining, and building upon deployments of industry-standard tools (e.g. GitLab, Docker, Bash)
  • Lead best-practices for building, testing, and releasing software including AI Services and DL models
  • Identifying infrastructure needs and translating them into action
  • Building tools for automatic content generation mechanisms that saves dozens of engineering hours

What we need to see:

  • BSc or MS degree in Computer Science, Computer Architecture or related technical field or equivalent experience
  • 5+ years of work experience in software development
  • Excellent Python programming skills, Great coding skills and a deep understanding of OOP concepts.
  • Familiarity with DevOps concepts such as CI/CD, Docker, Jenkins and automation tools
  • Experience in building both front-end (e.g. JS, React, Vue, Dash, Streamlit) and back-end services (e.g Flask, FastAPI, Django) services.
  • Understanding of Deep Learning allowing benchmarking on DL models
  • Willing to take action and strong analytical skills
  • Strong time-management and organization skills for coordinating multiple initiatives, priorities and implementations of new technology and products into very complex projects
  • Good communication and documentation habits

Ways to stand out from the crowd:

  • Experience with containerization technologies such as Docker
  • Experience in building monitoring or dashboarding solutions to support CI/CD pipelines
  • Hands-on in configuring complex CI pipelines
  • Experience with HPC based compute clusters and scheduling solutions like Slurm. Solid understanding of Linux environments

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most brilliant and forward-thinking people in the world working for us. If you're creative and autonomous, we want to hear from you! The base salary range is 144,000 USD - 270,250 USD. 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. NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

About Us

0:00 -/ - 3:32 NVIDIA is a Learning Machine
NVIDIA pioneered accelerated computing to tackle challenges no one else can solve. Our work in AI and the metaverse is transforming the world's largest industries and profoundly impacting society. Learn more about NVIDIA .

#J-18808-Ljbffr

Expertise level

Work arrangement

Similar Jobs in United States

Similar Jobs in