Software Engineer - ML Framework

Job summary

Responsibilities

Applied Intuition is pushing the frontier of autonomy and vehicle intelligence through next-generation AI Research. In this role, you will enable this research effort by building ML tools, infrastructure, managing large datasets, and addressing challenges across the entire end-to-end (E2E)…

• Build tools and infrastructure to enable large-scale E2E autonomy research led by Dr. Wei Zhan
• Work directly with both AI research and engineering teams to productionize next generation self-driving software
• Help scale end-to-end training by identifying bottlenecks, especially as we continue to scale our GPU compute, onroad data volume, curation efforts, and eval systems
• Technologies: Python, Pytorch, CUDA, Bazel, Kubernetes, Spark, Flyte

What they're looking for

• Exceptional software generalist experience with ambition to contribute up and down the ML stack, onboard the vehicle, and in the cloud
• Strong fundamental skills in software & module design: can tame chaos with libraries & abstractions
• Rapid analytical and problem-solving skills
• Bachelors, Masters or PhD degree in any field, or equivalent combination of education and work experience.
• 3+ years of experience in any field, or equivalent combination of education and experience

Tech stack

• Python
• PyTorch
• CUDA
• Kubernetes
• Spark