Staff Software Engineer, AI Libraries

Job Description

Design and deliver reusable Python libraries and platform tools for multimodal data loading, distributed training and inference, checkpointing, and observability across large GPU clusters. Lead projects end to end while improving system reliability, performance, developer experience, engineering standards, and support for machine learning teams.

Requirements: Requires deep Python expertise and a track record leading the design, delivery, and long-term evolution of high-quality software systems at scale. Candidates should be able to set technical direction, build stable reusable libraries, guide engineers, and have experience with concurrent, parallel, or distributed computing; familiarity with PyTorch or TensorFlow and cloud environments such as Azure is valuable.

Key Skills: Python, Software Architecture, Distributed Computing, Parallel Computing, Concurrent Computing, Machine Learning Platforms, PyTorch, TensorFlow, Azure, Multimodal Data Loading, Distributed Training, Inference, Checkpointing, Observability, API Design, Technical Leadership

Benefits: Equity, Relocation Support, Visa Sponsorship, Hybrid Working, Learning And Development Budget, Health Insurance, Dental Insurance, Enhanced Maternity And Paternity Leave, Retirement Or Pension, Therapist Access, Wellbeing Partnerships, Team Socials