Senior Machine Learning Researcher, Post-Training & Inference Optimization

THE ROLE: Fine-tune, distil and align large models for domain-specific tasks Build evaluation that measures what matters, not what is easy to measure Quantize, prune and optimize inference where latency carries a direct cost Write and profile custom GPU kernels when standard ops become the bottleneck Take research into production with engineering and trading teamsSTACK:Python: PyTorch, JAX, Num Py, PolarsGPU: CUDA, Triton, Nsight, NCCLC++17/20, pybind 11FSDP, Deep Speed or Megatron-style parallelismREQUIRED:A degree in a quantitative field from a leading university, with a minimum GPA of 3.5/4.0 Industry experience in ML roles, with hands-on post-training work Production-quality Python and C++GPU knowledge below the API: memory hierarchy, occupancy, and how to diagnose a slow kernel Eligibility to work in the USPREFERRED:A PhD or Master's in machine learning, computer science, mathematics, statistics or physics Leadership 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 lead You treat feedback as data and change course when it is right You 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).