Principal ML Systems Engineer - AI

Principal ML Systems Engineer - AI for QuantumLocation: Milpitas, California, United StatesJob Type: Full-Time (Senior Level, Research and Engineering Role)Responsibilities:Develop large-scale pretraining, fine-tuning, and inference systems for models used in quantum error correction (QEC) and classical control problems.Design, implement, and optimize GPU/CUDA/C++ kernels for high-performance model training and inference.Build distributed training pipelines for multi-node GPU clusters using frameworks like NCCL and Slurm.Collaborate with physicists and engineers to convert quantum simulation data into ML-ready representations.Prototype and evaluate new machine learning architectures for QEC-related optimization problems;Integrate ML systems into quantum computing pipelines in collaboration with infrastructure teams.Improve ML experimentation practices, including data curation, reproducibility, and evaluation frameworks.Develop and operate autonomous and agent-based AI systems for quantum architecture research.Qualifications:Strong software engineering skills in Python.Experience with PyTorch for training, fine-tuning, and deploying models.Experience with distributed training on GPU clusters (multi-node environments);Strong knowledge of GPU programming (CUDA, ROCm, or Triton) and memory optimization.Experience with ML infrastructure such as experiment tracking, model serving, and data pipelines.Experience building or working with autonomous or agentic AI systems.Preferred:Familiarity with quantum computing or quantum error correction.Experience in AI for Science domains such as physics simulation, materials science, or biology.Experience with multi-agent systems and orchestration frameworks.Rust or systems programming experience.Reinforcement learning in combinatorial or physical design spaces.Experience with hardware-adjacent systems such as FPGA toolchains or low-latency inference systems.Benefits and Compensation:Base salary range: $220,000 to $250,000 depending on location and experience.Equity participation for full-time employees.Comprehensive benefits package (details not fully specified in listing);Opportunity to work on utility-scale quantum computing systems targeting million-qubit architectures.Direct collaboration with leading researchers in quantum architecture and AI systems.Other Information:PsiQuantum focuses on silicon photonics-based quantum computing using scalable semiconductor manufacturing processes.Work involves direct collaboration with global partners and industry leaders in quantum computing development.Company is building fault-tolerant quantum computers for real-world industrial applications;Equal employment opportunity employer with strong commitment to diversity and inclusion.Candidates must ensure they apply only through official PsiQuantum channels.No visa sponsorship is available for this role.