Senior Software Engineer, AI Quantum

Position: Senior Software Engineer, AI for Quantum
Psi Quantum'smission is to build the first useful quantum computers-machines capable of delivering the breakthroughs the field has long promised. Since our founding in 2016, our singular focus has been to build and deploy million-qubit, fault-tolerant quantum systems.

Quantum computers harness the laws of quantum mechanics to solve problems that even the most advanced supercomputers or AI systems will never reach. Their impact will span energy, pharmaceuticals, finance, agriculture, transportation, materials, and other foundational industries.

Our architecture and approachisbased on silicon photonics. Byleveragingthe advanced semiconductor manufacturing industry-including partners like Global Foundries-we use the same high-volume processes that already produce billions of chips for telecom and consumer electronics. Photonics offers natural advantages for scale: photons don'tfeel heat, are immune to electromagnetic interference, and integrate with existing cryogenic cooling and standard fiber-optic infrastructure.

In 2024,Psi Quantum announced government-funded projects to support the build-out of our first utility-scale quantum computers in Brisbane, Australia, and Chicago, Illinois. These initiatives reflect a growing recognition that quantum computing will be strategically and economically defining-and that now is the time to scale.

Psi Quantum also develops the algorithms and software needed to make these systems commercially valuable. Our application, software, and industry teams work directly with leading Fortune 500 companies-including Lockheed Martin, Mercedes-Benz, Boehringer Ingelheim, and Mitsubishi Chemical-to prepare quantum solutions for real-world impact.

Quantum computing is not an extension of classical computing. Itrepresentsa fundamental shift-and a path to mastering challenges that cannot besolvedany other way. The potential is enormous, and we have a clearpathto make it real.

Come join us.

Job Summary

Psi Quantum is seeking an experienced Machine Learning (ML) expert to join our AI for Quantum initiative, focused on accelerating progress in Quantum Error Correction (QEC) and related computational research.

This role will partner closely with physicists and software engineers to develop, optimize, and scale ML models supporting quantum error correction and error modelling.

The ideal candidate brings deep expertise in large-scale model training and systems optimization, a solid mathematical or physical sciences foundation, and the ability to integrate advanced ML methods into a broader scientific software stack.

Responsibilities

• Develop large-scale pretraining, fine tuning and inference efforts for models used in QEC and related classical control problems.

• Design, implement, and optimize GPU/CUDA/C++ kernels and distributed training pipelines for high-performance model development.

• Collaborate with physicists and engineers to translate quantum phenomena and simulation outputs into ML-relevant data representations.

• Prototype and evaluate new ML architectures for solving QEC-related classical control problems.

• Work closely with infrastructure teams to ensure seamless integration of ML components into existing quantum computation pipelines.

• Drive best practices in ML experimentation, data curation, and reproducible research within the team.

Experience/Qualifications

• 8+ years of professional experience with a strong background in mathematics or physics (Bachelor's or higher in a quantitative field such as Physics, Applied Math, Computer Science, or related discipline).

• Proven expertise in large-scale pretraining, distributed optimization, and scaling methods (e.g., model/data parallelism).

• Fluent in Python and at least one systems programming language (C, C++, Rust).

• Proficiency in GPU programming and kernels (e.g. CUDA, Triton, Warp, CuTe, CUTLASS).

• Experience with GPU programming and performance profiling for ML workloads.

• Familiarity with modern ML frameworks such as PyTorch and JAX.

• Excellent communication skills and ability to collaborate with cross-disciplinary teams spanning physics, ML, and software engineering.

• Knowledge of model compression, distributed systems, or…