Machine Learning Engineer
Virtu Financial2 months ago
Base Salary
$200k - $300k/yr
Responsibilities
- Design and build experiment tracking and job orchestration infrastructure.
- Create tools for simulation lifecycle management, including back-tests and monitoring.
- Own visibility into GPU cluster utilization and track resource allocation.
- Diagnose and resolve performance issues in training pipelines.
- Build and maintain data pipelines for efficient data movement.
- Develop feature storage and retrieval patterns for training data access.
- Work with researchers to identify and reduce workflow friction.
- Collaborate on capacity planning and tooling decisions.
- Stay current with ML infrastructure developments and integrate valuable tools.
Requirements
- 5+ years of experience in ML engineering or research infrastructure.
- Strong Python engineering skills with clean, maintainable code.
- Experience with distributed training infrastructure and communication libraries.
- Practical experience with experiment tracking systems.
- Comfortable working across the Linux systems stack.
- Excellent communication skills for collaboration with researchers and engineers.
- Intellectually curious and self-driven in problem identification.