Machine Learning Specialist

Responsibilities

• Design and implement scalable and performant systems to serve machine learning models
• Optimize training and inference performance of deep learning models
• Create and maintain feature pipelines, feature stores, and model stores
• Monitor production models and setup retraining infrastructure to improve performance
• Optimize models for deployment on different hardware configurations
• Build frameworks to promote scalable and reproducible research
• Drive projects and lead new initiatives
• Facilitate discussions on code implementation, testing, and software architecture best practices
• Collaborate with stakeholders from different departments

Requirements

• Bachelor's or Master's degree in Computer Science, Software Engineering, or equivalent
• 5+ years of experience building software for machine learning systems
• Strong understanding of machine learning and deep learning concepts
• Excellent programming and scripting skills
• Experience building, validating, deploying, monitoring, and updating production models
• Experience with infrastructure and tooling for MLOps
• Experience with machine learning frameworks such as TensorFlow, PyTorch, TensorRT, or ONNX
• Experience with GPU programming using CUDA, OpenCL, or similar libraries
• Experience optimizing inference performance via quantization, profiling, or kernel tuning

Preferred Qualifications

• Experience with distributed systems
• Multi-threaded programming experience
• Knowledge of hardware/software interaction

About the Company

DRW is a diversified trading firm with over 3 decades of experience bringing sophisticated technology and exceptional people together to operate in markets around the world.