Senior and/or Principal Software Engineering- AI Frameworks
Core
Design, implement, test, and operate production-quality components across AI frameworks, runtimes, benchmarking systems, and performance tooling to accelerate model and hardware onboarding.
Role type
Senior/Principal Software Engineer (AI Frameworks & Infrastructure)
Builds
Production-ready AI framework components, automation, observability systems, and performance tooling
Domain
Artificial Intelligence / Machine Learning Infrastructure
Deliverable
production ML models | infrastructure
Required skills
C++ or Python, performance analysis, benchmarking, automation, AI/ML frameworks (PyTorch, TensorFlow, ONNX Runtime), GPU software (CUDA, ROCm, Triton), collective operations, networking concepts, software architecture, computer architecture, accelerator-aware optimization
Preferred skills
Hands-on technical leadership, cross-team collaboration, technical ownership, creating reusable platforms, influencing senior stakeholders, mentoring technical leaders
Technologies
PyTorch, TensorFlow, ONNX Runtime, CUDA, ROCm, Triton
Responsibilities
Design and implement production-quality components for AI frameworks and runtimes; Build automation and observability to detect regressions and improve reproducibility; Partner with researchers and infrastructure owners to diagnose cross-stack issues; Contribute to technical design reviews and engineering standards; Provide hands-on technical leadership through prototypes and complex debugging
Seniority
Senior/Principal, hands-on IC with leadership responsibilities