Software Developer - ML and GenAI (Hybrid)

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 a ML Developer at Xanadu, you will play a crucial role in integrating generative AI into our R&D methodologies. This unique position allows you to contribute to the development of our internal tool stacks, aimed at advancing the first commercially viable quantum computer. The ML team’s focus is to build and improve the modeling, optimization, simulation, data processing, and design methodology for our internal research. Situated at the intersection of multiple technical disciplines, you will have the unique opportunity to work with world leading researchers, scientists, engineers, and software developers. By infusing cutting-edge ML into the software tools, you have the chance to super-charge their research progress and even reshape how they work. You will:

• Design and develop ML tooling softwares for accelerating internal research.
• Build processes and pipelines for structuring unstructured documents at scale.
• Collaborate closely with theoretical quantum computer researchers to craft new ML-driven solutions for complex challenges.
• Explore and experiment with integrating GenAI/LLMs into the ML tool stack.
• Enhance and scale up existing optimization and simulation algorithms.
• Develop SDKs, APIs, and data visualizations that interface with internal users.
• Ensure high-quality production code through automated tests and a dynamic code review process.

At Xanadu, we primarily work with Python, numpy/jax/torch/tensorflow, GitHub, Docker, CI pipelines, and multiple cloud platforms. Proficiency in these technologies is essential.

Required qualifications and experience:

• BSc in Computer Science, Engineering, Physics, Math, or a related field.
• 2+ years of experience building machine learning software stacks in a research environment that involve at least a subset of LLM, generative AI, deep NN design/training/deployment, optimization, simulation, and data ETL.
• Strong knowledge of Python and cloud computing.
• Rich experience in large scale data wrangling from unstructured documents.
• Hands-on experience with generative AI, prompting techniques, knowledge retrieval (RAG), and agentic systems.
• Experience with ML assisted data processing, labeling, and distillation.
• Experience systematic validation and benchmarking of LLMs.
• Familiar with working with multidimensional arrays in Numpy, PyTorch, Tensorflow, Jax, etc.
• Good and up-to-date understanding of ML concepts.
• Knowledge of the full lifecycle of software development, including version control, code review, testing, CI/CD, logging, profiling, debugging, and documentation. 
• Adept at working with Linux shell, Git, and GitHub.
• Strong analytical and problem-solving skills.
• Curiosity and enthusiasm about new and unfamiliar technologies with ability to learn technical/scientific concepts quickly with minimal direction.
• Strong communication and collaboration skills.
• Basic knowledge of physics and linear algebra.

Preferred qualifications and experience

• MSc in Computer Science, Engineering, Physics, Math, or related field.
• Experience building tooling software in a research environment.
• Experience building large scale ML systems.
• Good knowledge in physics and linear algebra.
• Familiarity with distributed computing.
• Familiarity with optimization algorithms.
• Familiarity with statistical analysis/modeling.
• Familiarity with quantum computing.