AI Specialist - GPU Computational Physics

AI Specialist - GPU Computational Physics

About Xanadu: 
Xanadu’s mission is to build quantum computers that are useful and available to people everywhere. 

At Xanadu, we are learners, innovators, researchers, collaborators and problem solvers. We are creating something that has never been built before. What we are doing is extremely hard, the classic moon shot. Few people in their life will be able to be a part of something like this, where if we are successful, the technologies we develop will solve some of the world’s most challenging problems, and literally change the world. And that is something to be excited about!

Your role and responsibilities: 
As an AI Specialist at Xanadu, you will drive applied AI initiatives by accelerating the very core of our R&D efforts. Your work will focus on enhancing the performance of our most crucial systems, from the design and characterization of our quantum hardware to the execution of internal simulation software on high-performance computing platforms. Using state-of-the-art machine learning, GPU programming, and computational physics techniques, you will accelerate our progress toward the first commercially viable quantum computer. The AI team focuses on building and improving modeling, simulation, optimization, data processing, and design methodology for all internal research. At the intersection of multiple technical disciplines, you will collaborate with leading researchers, scientists, engineers, and software developers, using cutting-edge computational methods and ML to enhance software tools and potentially transform research processes.

You will:

• Develop and implement GPU-accelerated numerical methods, including ODE/PDE solvers and Monte Carlo methods, combined with statistical analysis of simulation uncertainties.
• Optimize the execution of internal simulation software by leveraging GPU programming, custom kernels, code profiling and optimization, and high-performance computing (HPC) platforms.
• Design and build high-performance simulators for complex physical systems, often modeled by differential equations.
• Develop and rigorously test new computational methods, ML algorithms, and toolkits to improve R&D efficiency.
• Collaborate closely with hardware engineers and scientists to create and implement novel simulation and ML-driven solutions for complex research challenges.
• Establish and maintain reproducible data analysis and modeling workflows.

At Xanadu, we primarily work with Python, JAX, GitHub, Docker and multiple cloud platforms. Proficiency in these technologies is essential.

Basic qualifications and experience:

• BSc. in Physics, Math, Computer Science, Engineering, or a related field.
• 6+ years of industry experience in any of the fields: computational physics, scientific computing, simulation software, physical sciences, ML/AI, data science, or control systems, etc.
• Strong knowledge of Python and its numerical/scientific ecosystem (NumPy, xarray, PyTorch, JAX, SciPy, scikit-learn, Ray, etc.).
• Hands-on experience with GPU acceleration and performance optimization of numerical or simulation software.
• Experience writing and designing simulators for complex physical systems, using methods such as ODE/PDE solvers or Monte Carlo simulation.
• Experience with GPU programming (e.g., CUDA), kernel development, and GPU computing frameworks such as JAX.
• Experience with software development lifecycles, including version control, code review, testing, CI/CD, logging, profiling, debugging, and documentation.
• Comfortable working with Linux shell, Docker, Git, and GitHub.
• Enthusiasm for learning new technologies and scientific concepts with minimal supervision.
• Solid communication and collaboration skills.
• Strong self-driven analytical and problem-solving abilities.
• Good knowledge of physics, numerical methods, linear algebra, and statistics.

Preferred qualifications and experience:

• MSc/PhD in Computer Science, Engineering, Physics, Math, or related field.
• Excellent knowledge in physics and linear algebra.
• Experience with scientific machine learning, differentiable simulation, or generative modeling.
• Familiarity with or curiosity towards quantum computing.
• Rich experience in any of these areas:
• Statistical inference and uncertainty quantification
• GPU programming, custom CUDA kernels, and performance tuning
• Large-scale simulations of complex physical systems
• Numerical methods for ODEs/PDEs or Monte Carlo simulation

This is for a new position. Your base salary will be determined based on your location, experience, and internal benchmarks. The base salary range is 140,000 - 190,000 CAD. You will also be eligible for equity and benefits.

Our values are important. They are fundamental and lay the foundation for culture at Xanadu. Learn more about our values here.

We are an equal opportunity employer and encourage candidates of all backgrounds to apply. We are committed to building an inclusive, safe, and equitable culture and fostering an environment where our employees feel included, valued, and heard. We are committed to meeting the needs of all individuals and support a barrier-free workplace. Should you require accommodations at any point during the recruitment process please contact Recruiting at [email protected].

Please be advised that we may use artificial intelligence (AI) tools to assist in the screening and assessment of applicants for this position. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us.