AI Specialist - Representation and Reinforcement Learning

Xanadu; Toronto; On-site; CAD 140,000 - 190,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

Xanadu is seeking an AI Specialist to drive applied AI initiatives across diverse R&D data. You will use state-of-the-art ML techniques to uncover insights and build internal tool stacks for advancing the first commercially viable quantum computer.

The role involves collaborating with researchers, engineers, and software developers, focusing on representation learning, reinforcement learning, and generative models. Proficiency in Python, Jupyter, Docker, and cloud pipelines is essential.

Qualifications

• Bachelor’s degree in a relevant field and 4+ years in deep learning/AI/ML.

Responsibilities

• Analyze complex structured and unstructured R&D data to identify trends and insights.
• Develop representation/reinforcement/generative learning strategies for diverse data.
• Design and implement ML and optimization algorithms for R&D problems.
• Develop and test new ML toolkits to improve R&D efficiency.
• Collaborate with hardware engineers and scientists to create ML solutions.
• Establish reproducible data analysis and modeling workflows.

Skills

Python programming

Deep learning / ML

Data analysis

Communication

Mathematics / linear algebra

Self-driven / analytical

Education

BSc in Physics/Math/CS/Engineering

MSc/PhD preferred

Tools

Docker

Git

GitHub

Jupyter

Jax

CUDA / PyTorch stack

Job description

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 deeply analyzing diverse R&D data and processes. Using state-of-the-art machine learning and AI techniques, such as representation learning, generative modeling, reinforcement learning etc., you will uncover hidden patterns in research across various technical fields. This work will directly contribute to developing internal R&D tool stacks to advance 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 AI/ML to enhance software tools and potentially transform research processes. You will:

• Investigate and analyze complex structured and unstructured data from various internal R&D projects to identify key trends and insights.
• Develop generalizable representation/reinforcement/generative learning strategies for diverse research data, addressing both theoretical and engineering challenges.
• Design and implement machine learning and optimization algorithms based on learned representations to solve specific R&D problems.
• Develop and rigorously test new ML algorithms and tool kits to improve R&D efficiency.
• Collaborate closely with hardware 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, Jupyter, Jax, GitHub, Docker, CI pipelines, 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.
• 4+ years of industry experience in deep learning/AI/ML, including at least one of these topics: representation learning, reinforcement learning, geometric deep learning, computer vision, NLP, generative models, GFlowNet, control theory.
• 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
• Experience with designing and building novel and generalizable representations of complex data structures with symmetries.
• Hands-on experience with large scale training of neural networks for RL, LLMs, diffusion models, or other types of generative modeling.
• 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 and linear algebra.

Preferred Qualifications And Experience

• MSc/PhD in Computer Science, Engineering, Physics, Math, or related field.
• Excellent knowledge in physics and linear algebra.
• Familiarity with or curiosity towards quantum computing.
• Rich experience in any of these areas:
  • GFlowNet
  • Geometric deep learning and equivariant models.
  • ML on ultrafast embedded systems
  • Modeling and simulation of physical systems on high performance computing hardware
  • Training of commercial grade LLMs

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 [e-mail].

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.