Principal ML Systems Engineer – AI

Job Description

Job Title: Principal ML Systems Engineer – AI for Quantum. Location: Milpitas, California, United States

Job 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.