Staff Software Engineer, AI Foundation Model

8+ years of software engineering experience, with a substantial portion focused on state of the art and next-gen AI model development, ML infrastructure, MLOps, data engineering, or AI platform development.

Experience in development of the AI-first Autonomy software tech stack that includes Perception, Behavior Planning, Prediction and Actuation ( based on Transformer architectures)

Demonstrated ability to design and deliver large-scale, production-grade systems independently — from initial architecture through deployment and operation.

Deep proficiency in Pytorch, TensorFlow and at least one systems language such as C++,C, Rust, or Go.

Hands-on experience with distributed training frameworks (Python extensions, JAX, or equivalent) and the infrastructure required to run them reliably at scale.

Experience in onboarding trained models into edge environments consisting of High performance compute System-on-Chips.

Strong background in data pipeline design, including streaming and batch processing, data versioning, and handling high-volume, heterogeneous sensor data.

Experience building model evaluation and validation frameworks leveraging MLOps frameworks (like Weights and Biases. MLFlow etc) that go beyond accuracy metrics to assess real-world reliability and safety-relevant behavior.

Proven track record of influencing technical direction across teams and driving alignment on shared infrastructure and standards.

Excellent written communication skills, with the ability to produce clear design documents and architecture proposals that hold up to rigorous review.

Comfort working in a fast-moving startup where the scope of problems and priorities evolve as the product matures.