Principal ML Systems Engineer – AI for Quantum
PsiQuantum20 days ago
Base Salary
$220k - $245k/yr
Responsibilities
- Develop large-scale pretraining, fine-tuning, and inference efforts for models used in quantum error correction.
- Design, implement, and optimize GPU/CUDA/C++ kernels and distributed training pipelines.
- Collaborate with physicists and engineers to translate quantum phenomena into ML-relevant data representations.
- Prototype and evaluate new ML architectures for solving classical control problems.
- Work closely with infrastructure teams to integrate ML components into quantum computation pipelines.
- Drive best practices in ML experimentation, data curation, and reproducible research.
Requirements
- Strong software engineering fundamentals in Python.
- Experience with PyTorch for training and deploying models in production.
- Knowledge of distributed training on GPU clusters.
- Experience in GPU kernel development using CUDA, ROCm, or Triton.
- Familiarity with AI/ML infrastructure including experiment tracking and data pipelines.
- Experience building or operating autonomous or agentic AI systems.