Machine Learning Engineer

Virtu’s Research Technology team is looking for an experienced Machine Learning Engineer to join a small group of technologists whose primary function is building the infrastructure that powers our quantitative researchers. This is a unique opportunity to work at the intersection of machine learning and systematic trading — building tools that directly determine how fast our researchers can move, and how effectively our GPU cluster translates into research output.In this role, you will be responsible for the development of our ML research platform: the systems that manage data and compute, track experiments, and enable researchers to go from idea to result as efficiently as possible. You will work closely with quants and engineers alike and will play a central role in shaping how ML is done at the firm as we scale our capabilities. We mostly use Python, C++ and Java with a variety of open-source tools along with proprietary solutions.THE ROLEDesign and build experiment tracking, job orchestration, and reproducibility infrastructure so researchers can iterate quickly, compare runs reliably, and recover from failures without losing workCreate tools for all stages of the simulation lifecycle including historical back-tests and production monitoring. Add new features to our simulatorsOwn visibility into GPU cluster utilization — track allocation, surface bottlenecks, and ensure our compute investment is being used effectivelyDiagnose and resolve performance issues across training pipelines: data loading throughput, storage I/O, GPU utilization, and inter-node communication in distributed training runsBuild and maintain data pipelines that move financial data from storage into training workflows efficiently, with strong guarantees on correctness and versioningDevelop feature storage and retrieval patterns that support fast, reproducible access to training data at scaleWork directly with researchers to understand friction in their workflows, and build solutions that reduce it — from tooling improvements to infrastructure changesCollaborate with existing infrastructure engineers on capacity planning, cloud/on-prem tradeoffs, and tooling decisions — this is a collaborative environment, not a siloed oneStay current with developments in ML infrastructure tooling and bring relevant ideas and tools into our stack where they create genuine valueTHE CANDIDATE5+ years of experience in ML engineering, research infrastructure, or HPC environmentsStrong Python engineering skills — you write clean, maintainable, well-tested code that other engineers want to build on. Exposure to C++ in a performance-sensitive context is a plusExperience building or operating distributed training infrastructure, with working knowledge of how collective communication libraries (NCCL, Horovod, or similar) behave at scalePractical experience with experiment tracking systems and strong opinions about what good research infrastructure looks likeComfort working across the Linux systems stack — storage, networking, job scheduling — enough to follow a problem wherever it leadsExcellent communication skills and the ability to work closely with researchers and engineers across disciplinesIntellectually curious and self-driven — you proactively identify problems worth solving, not just problems you've been asked to solveDESIRED, BUT NOT REQUIREDExperience with on-prem compute environments and job orchestration tools such as SlurmFamiliarity with GPU profiling tools (NSight Systems, PyTorch Profiler) and hands-on experience optimizing GPU memory or compute utilizationExperience with columnar data formats and high-performance data processing tools such as Parquet, Arrow, and PolarsFamiliarity with workflow orchestration tools (Prefect, Dagster, or similar)Prior experience in environments with high-stakes, time-series data at scale. Open to Quantitative Finance, Algorithmic Trading, and OtherExperience contributing to or extending open-source ML frameworks or infrastructure toolingSalary Range: $200,000 - $300,000 (salary range is exclusive of bonuses, benefits or other categories of compensation)Virtu Financial is an equal opportunity employer, committed to a diverse and inclusive workplace, welcoming you for who you are and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation, protected veteran status, disability, age, or other legally protected status.