Senior AI/ML Software Engineer
Core
Build and maintain tools and infrastructure to automate functional and performance validation of deep learning libraries and models across ROCm and GPU platforms.
Role type
Senior IC software engineer (ML infrastructure & validation)
Builds
Automated testing pipelines, benchmarking tools, and real-time performance dashboards for AI/ML frameworks
Domain
Semiconductor / AI Infrastructure / GPU Computing
Deliverable
production ML models
Required skills
Python, Linux, CI/CD pipeline design, automated testing frameworks, performance tuning, debugging, version control, containerized deployments
Preferred skills
Machine learning frameworks (PyTorch, TensorFlow), cluster orchestration (SLURM, K8s), profiling tools, regression tracking systems
Technologies
Python, Linux, GitHub Actions, Buildkite, ROCm, Docker, PyTorch, TensorFlow, SLURM, Kubernetes
Responsibilities
Design and manage CI/CD pipelines for rapid software deployment; Integrate automated testing frameworks into CI pipelines; Automate functional and performance testing of AI libraries and models; Develop tools for continuous benchmarking and regression tracking; Build and maintain real-time dashboards for performance metrics; Collaborate with internal teams to support diverse model workloads; Contribute to open-source repositories and Docker releases
Seniority
Senior, hands-on IC
