Senior and/or Principal Software Engineer- AI Frameworks

About the job

As a Senior Software Engineer: Design, implement, test, and operate production-quality components across AI frameworks, runtimes, benchmarking systems, performance tooling, and service integrations. Build automation and observability that detect regressions, improve reproducibility, surface actionable insights, and accelerate model and hardware onboarding. Drive scoped projects from problem definition through deployment, balancing delivery speed, maintainability, reliability, and measurable customer or capacity impact. Partner with researchers, model teams, infrastructure owners, and hardware vendors to diagnose cross-stack issues and deliver production-ready solutions. Contribute to technical design reviews, engineering standards, operational health, and mentoring of other engineers. Establish common measurement, automation, observability, and engineering mechanisms that turn one-off analyses into scalable platform capabilities. Provide hands-on technical leadership through prototypes, critical-path implementation, design and code reviews, complex debugging, and operational readiness. Bachelor's Degree in Computer Science or a related technical field and 4+ years of technical engineering experience coding in languages such as C++, or Python, or equivalent experience. Experience building and operating complex software systems Practical knowledge of performance analysis, benchmarking, automation, or developer tooling Familiarity with AI/ML frameworks such as PyTorch, TensorFlow, or ONNX Runtime Familiarity with GPU software and profiling technologies such as CUDA, ROCm, Triton, or equivalent Demonstrated cross-team collaboration and technical ownership Extensive experience designing and shipping complex, high-performance or distributed software systems Experience in software architecture, computer architecture, and accelerator-aware optimization Experience with AI/ML workloads and frameworks Demonstrated leadership of cross-team technical initiatives from strategy through production Track record of creating reusable platforms, influencing senior stakeholders, and mentoring technical leaders