AI Specialist - Applied ML Research

Skip To Job Description AI Specialist - Applied ML Research Toronto, ON, Canada Full Time AI/ML Mid Level Share 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. 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 optimizing the core of our R&D efforts and deeply analyzing diverse R&D data and processes. 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, AI techniques, GPU programming, and advanced optimization, you will accelerate our progress toward the first commercially viable quantum computer. The AI team focuses on building and improving modeling, optimization, simulation, 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 ML to enhance software tools and potentially transform research processes.You will:Develop generalizable representation/reinforcement/generative learning strategies for diverse research data, addressing both theoretical and engineering challenges.Develop and implement ML-based design optimization strategies for hardware and software systems.Investigate and analyze complex structured and unstructured data from various internal R&D projects and products to identify key trends and insights.Continuously monitor, evaluate, and incorporate state-of-the-art ML and AI developments into internal R&D tool stacks to maintain a competitive advantage in quantum hardware progress.Collaborate closely with engineers and scientists to create and implement novel 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, CI pipelines, Jupyter 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.5+ years of industry experience in ML/AI/deep learning, data science, statistics, physical sciences, physical design and optimization, simulation software, or control systems.Strong knowledge of Python and its numerical/scientific ecosystem (jax, numpy, pandas, xarray, pytorch, cuda, scipy, sklearn, ray, etc.).Deep mathematical understanding of machine learning and optimization.Hands-on experience in one or more of the following focus areas:ML-based optimization and the optimization of numerics or simulation software.Deep learning/AI/ML specializations such as representation learning, reinforcement learning, geometric deep learning, computer vision, NLP, generative models, GFlowNet, or control theory.Designing and building novel and generalizable representations of complex data structures with symmetries.Large-scale training of neural networks for RL, LLMs, diffusion models, or other types of generative modeling.Writing and designing simulators for complex physical systems (e.g., differential equations) and control systems.Experience with GPU programming (e.g., CUDA).Experience with software development lifecycles, including version control, code review, testing, CI/CD, logging,...