What are the responsibilities and job description for the Junior Software Engineer, Machine Learning Infrastructure position at Jobright.ai?
Verified Job On Employer Career Site
Job Summary:
Nuro is a robotics start-up focused on enhancing everyday life through robotics and autonomous driving technology. The role involves building scalable machine learning infrastructure and distributed training solutions to accelerate ML development, contributing to advancements in machine learning research.
Responsibilities:
• Build up a model serving platform for efficient large-scale simulations and reinforcement learning (RL) training.
• Maintain observability and monitoring for critical services like ML training, data dumping, and deployment.
• Implement tools to track the model development lifecycle for an efficient deployment and evaluation process.
Qualifications:
Required:
• 2 years of relevant work experience or an equivalent experience in Masters/PhD with 1 years of relevant experience.
• Strong coding, software design, and debugging skills in Python or C .
• You have relevant exposure to the ML development life cycle and ML models.
• You are highly productive, motivated, and are a strong team player.
• You thrive in complex, fast-paced environments and learn quickly by doing.
Preferred:
• You have experience in building a cloud-based distributed training platform.
• You have experience profiling and optimizing performance bottlenecks for deep learning models and trainers.
• Experience with serving frameworks like Triton or VLLM
• Experience with model & data parallel training frameworks like PyTorch FSDP
Company:
Nuro is a robotics company specializing in the development of autonomous driving technologies. Founded in 2016, the company is headquartered in Mountain View, California, USA, with a team of 501-1000 employees. The company is currently Late Stage. Nuro has a track record of offering H1B sponsorships.