Senior Machine Learning Researcher, Post-Training & Inference Optimization

A top-tier global trading firm is building out its post-training research. The mandate is to take large pretrained models and adapt them to financial markets. Those models have to deal with noisy data, distributions that shift daily, and competitors who adapt to any edge. They also have to run within tight latency budgets. Performance is measured against real market outcomes rather than benchmarks or human raters. The firm trades at a scale few can match, so small model improvements have a measurable effect within days. You will join a small group of researchers whose work goes straight into production.THE ROLE:Fine-tune, distil and align large models for domain-specific tasksBuild evaluation that measures what matters, not what is easy to measureQuantize, prune and optimize inference where latency carries a direct costWrite and profile custom GPU kernels when standard ops become the bottleneckTake research into production with engineering and trading teamsSTACK:Python: PyTorch, JAX, NumPy, PolarsGPU: CUDA, Triton, Nsight, NCCLC++17/20, pybind11FSDP, DeepSpeed or Megatron-style parallelismREQUIRED:A degree in a quantitative field from a leading university, with a minimum GPA of 3.5/4.0Industry experience in ML roles, with hands-on post-training workProduction-quality Python and C++GPU knowledge below the API: memory hierarchy, occupancy, and how to diagnose a slow kernelEligibility to work in the USPREFERRED:A PhD or Master's in machine learning, computer science, mathematics, statistics or physicsLeadership experience within ML roles, such as technical lead, mentoring or owning a workstreamHOW YOU WORK:You start on problems without waiting to be asked, and you know when to bring in your leadYou treat feedback as data and change course when it is rightYou want to be the least experienced person in the room some days, because that is where the learning isPRE APPLICATION:Please do not apply if you're looking for a contract or remote work.You must be eligible to live and work in the US, without requiring sponsorship.Please ensure you meet the required experience section prior to applying.Allow 1-5 working days for a response to any job enquiry.Your application is subject to our privacy policy (thurnpartners.com/privacy-policy).