AI Engineer

As an AI Engineer at DRW, you will work on high-impact machine learning and artificial intelligence initiatives within a collaborative multi-asset trading team. You will build critical infrastructure and tooling that leverages AI and ML technology to enhance productivity and efficiency. You will drive projects from initial proof-of-concept to production deployment, create feature pipelines and model stores, implement automated systems for continuous training and monitoring, and troubleshoot performance bottlenecks to optimize GPU and CPU resource usage. Surrounded by cutting-edge technology and experienced trading teams, you will tackle fascinating new projects while becoming a subject-matter expert in a critical and innovative role.

Responsibilities

• Drive end-to-end development of AI infrastructure and AI-driven applications from proof-of-concept to production deployment and maintenance
• Collaborate with technologists, traders, quantitative researchers, and data scientists to identify high-impact AI and ML integration opportunities
• Implement automated systems for continuous training, validation, and monitoring of models
• Provide technical leadership in selecting, integrating, and optimizing AI and ML frameworks, libraries, and tools
• Create and maintain feature pipelines, feature stores, and model stores
• Develop frameworks to enable scalable, reproducible research
• Troubleshoot performance bottlenecks, conduct root-cause analyses, and optimize GPU or CPU resource usage

Requirements

• Bachelor's or advanced degree in Computer Science, Machine Learning, Artificial Intelligence, or a related field
• 3+ years of experience working with machine learning and artificial intelligence technology
• Strong understanding of core machine learning and artificial intelligence concepts
• Excellent programming skills in Python
• Demonstrated experience building, validating, deploying, monitoring, and updating production ML and AI models
• Hands-on experience with MLOps and AIOps infrastructure and tooling
• Proficient in problem-solving and analytical reasoning
• Exceptional communication and collaboration skills
• Experience with ML frameworks such as TensorFlow, PyTorch, TensorRT, or ONNX
• Experience with Large Language Models, including RAG and fine-tuning techniques
• Familiarity with compute infrastructure necessary to support AI and ML technology

Benefits

• Group medical, pharmacy, dental and vision insurance
• 401k with discretionary employer match
• Short and long-term disability insurance
• Life and AD&D insurance
• Health savings accounts
• Flexible spending accounts
• Annual discretionary bonus