Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff Engineer: Scalable ML Training & Distributed Systems

Reflection

Reflection is seeking a software engineer to design and optimize the core training infrastructure powering our models. You will work on RL training loops, distributed GPU systems, and large-scale data pipelines, collaborating with researchers to translate ideas into reliable, scalable systems. You will tackle bottlenecks, implement state-of-the-art techniques, and build tooling for launching, monitoring, and reproducing experiments, with a focus on performance, stability, and reproducibility. #J-18808-Ljbffr Reflection

Vacancy posted 5 days ago
Similar jobs that could be interesting for youBased on the Staff Engineer: Scalable ML Training & Distributed Systems in New York, NY vacancy
  • $170k - $225k

     ...company with employees distributed across the US and EU...  ...we’re looking for a Staff Machine Learning Engineer to take technical...  ...of our core ranking system. Every job request on...  ...most consequential ML systems we run.This...  ...to produce cleaner training data and unlock downstream... 
    Training
    Work at office
    Immediate start
    Flexible hours

    Taskrabbit

    New York, NY
    6 days ago
  • Outtake is seeking a staff-level infrastructure generalist to set...  ...technical direction of our systems. You will own the foundations...  ...rapid growth. You’ll design scalable asynchronous processing, data...  ...ingestion, and reusable services for engineers, while guiding migrations and... 
    Suggested
    Work at office

    Outtake

    New York, NY
    2 days ago
  • $265k - $331k

     ...through frontier AI systems that solve real...  ...of the hardest engineering challenges.As a Staff Frontier Agent...  ...Unlike traditional ML roles that focus...  ...solutions into scalable production...  ...EngineeringExperience building distributed production...  ...education or training. Scale employees... 
    Training
    Full time

    Scale AI

    New York, NY
    5 days ago
  • $210k - $234k

     ...buyer clients. Engineering @ Compass Compass...  ..., resilient, and scalable web apps, mobile...  ...data retention, and training-use terms of third...  ...complex AI systems at scale. You have...  ...closing gaps in AI/ML workflows through...  ...modeling complex distributed systems and driving... 
    Training
    Minimum wage
    Flexible hours

    Compass, Inc.

    New York, NY
    5 days ago
  • $180k - $250k

     ...run demanding training, inference, and...  ...teams build the systems needed to operate...  ...software engineering to help engineers...  ...the underlying ML systems, services...  ...About the roleAs a Staff Machine...  ...maintainable, scalable services with clear...  ...to reason about distributed systems, data quality... 
    Training
    Permanent employment
    Full time
    Temporary work
    Casual work
    Work at office
    Flexible hours

    CoreWeave

    New York, NY
    3 days ago
  • $170k - $225k

     ...with employees distributed across the US and...  ...re looking for a Staff Machine Learning Engineer to join our team...  ...the robust systems that power repeat...  ...loyalEnd-to-End ML Lifecycle: Own the...  ...engineering and training through evaluation...  ...& Scalability: Build and maintain... 
    Training
    Work at office
    Immediate start
    Flexible hours

    Taskrabbit

    New York, NY
    3 days ago
  •  ...Staff Machine Learning Engineer We are seeking an exceptional Staff...  ...architect large-scale ML systems that detect and...  ...learning expertise with scalable engineering and...  ...engineering, model training, deployment, and monitoring...  ...with big data and distributed systems (e.g.,... 
    Training

    AppGate

    New York, NY
    1 day ago
  • Staff Security Engineer Assured is on a mission to modernize...  ...large-scale ML-driven decision-making...  ...risk, designing scalable security...  ...attacker—identify systemic weaknesses and design...  ..., playbooks, and training programs that make...  ...SaaS architectures, distributed systems, and... 
    Training
    Temporary work
    Remote work
    Work from home
    Home office

    ASSURED

    New York, NY
    2 days ago
  • $144.25k - $256.25k

     ...benefitsJob Function: Engineering &...  ...companySenior Staff AI Engineer leads...  ...and agentic AI systems. The role sets...  ...direction, ensures scalable, reliable, secure...  ...architectures, training, evaluation, and...  ...foundation in distributed systems, cloud-...  ...experience in AI/ML engineering, AI... 
    Training
    Visa sponsorship

    American Express

    New York, NY
    6 days ago
  •  ...at marketplace scale. In this Staff Machine Learning Engineering role, you will set technical direction...  ...and deliver end-to-end ML solutions that support distributed systems handling billions of trips....  ...functional partners, and drive scalable ML models from #J-18808-Ljbffr... 

    Uber

    New York, NY
    13 hours ago
  •  ...company’s technology engineering, architecture,...  ...Senior Staff AI Engineer leads...  ...and agentic AI systems. The role sets...  ...direction, ensures scalable, reliable,...  ...architectures, training, evaluation, and...  ...foundation in distributed systems, cloud-...  ...experience in AI/ML engineering, AI... 
    Training
    Work experience placement
    Visa sponsorship

    Aimlroles

    New York, NY
    1 day ago
  • $192k - $260k

     ...their business. Founded by engineers — and customer obsessed...  ...bring enterprise-grade ML and AI personalization...  ...customer problems into scalable, trustworthy...  ...evaluating ML models and/or LLM systems for real product or...  ...relevant certifications and training, and specific work... 
    Training
    Work at office
    Local area
    Worldwide

    DataBricks

    New York, NY
    4 days ago
  • $97.4k - $174.05k

     ...the organization with scalable capabilities for...  ...industries.You'll join a 100-engineer remote-first team...  ...and ship full-stack systems, shape platform...  ...Impact: Our globally distributed team of ~100 engineers...  ...on traditional ML tasks like training, deployment, and monitoring... 
    Training
    Full time
    Work at office
    Remote work
    2 days per week

    Wolters Kluwer

    New York, NY
    6 days ago
  • $245k - $319k

     ...are looking for a Senior Staff Software Engineer, Machine Learning to be...  ...the next generation of AI/ML models and systems that connect buyers and...  ...remain high-performance and scalable. We are proud of our...  ...on both large-scale model training and serving. ~ A... 
    Training
    Full time
    Work at office
    Local area
    Remote work
    Visa sponsorship
    Flexible hours

    Etsy, Inc.

    New York, NY
    1 day ago
  •  ...We’re looking for a Staff Machine Learning Engineer I to join the Search Retrieval...  ...to build efficient, scalable, and intelligent retrieval systems that surface the most...  ...pipelines through training, evaluation, and production...  ...about using modern ML to improve search... 
    Training
    Full time
    Work at office
    Local area
    Immediate start
    Remote work
    Visa sponsorship

    Etsy

    New York, NY
    1 day ago
  • $229k - $343k

     ...Staff Machine Learning Engineer, Diffusion, Generative Modeling and Inference...  ...Snap’s Generative ML Platform team builds...  ...power creative, scalable experiences for hundreds...  ..., and agentic systems that empower creators...  ...field Experience training large-scale diffusion... 
    Training
    Live in
    Work at office
    Local area
    Worldwide

    Snap Inc.

    New York, NY
    1 day ago
  •  ...to accelerate how the world engineers new hardware. Nominal's connected...  ...team spans data science and ML, distributed systems, search, and knowledge...  ...them. The Role: As a Staff Research Engineer supporting agent evals & post-training, you'll define how Nominal measures... 
    Training
    Permanent employment

    Nominal

    New York, NY
    3 days ago
  • $227.2k - $324.5k

    About the Role:As a Staff Software Engineer on the ML Infrastructure team,...  ...latency ML model serving systems that support Deep...  ...:Design and build scalable, high throughput, and low latency distributed systems using ScalaBuild...  ...ElastiCache, model training orchestration, etc.... 
    Training
    Full time
    Temporary work
    Local area
    Flexible hours

    Tubi TV

    New York, NY
    6 days ago
  • $260k - $274k

     ...building the operating system for value...  ...architecture, reliability, scalability, governance, and...  ...looking for an engineer who can move...  ...will do As a Staff Data Engineer on...  ...data platforms, distributed and stream processing...  ...~$1,000 Annual training budget: we support... 
    Training
    Work at office
    Immediate start
    Remote work
    Home office
    Flexible hours

    MoonPay

    New York, NY
    1 day ago
  •  ...platform and product engineering teams Threat model AI...  ..., data retention, and training-use terms of third-party...  ...securing complex AI systems at scaleYou write production...  ...closing gaps in AI/ML workflows through...  ...threat modeling complex distributed systems and driving remediation... 
    Training

    Compass

    New York, NY
    1 day ago
  • $229.9k - $262.4k

     ...Senior Lead AI Engineer (Gen AI Platform Services: Distributed Systems) Overview: At Capital One,...  ...our applications of AI & ML are bringing humanity and...  ...product experiences and scalable, high-performance AI infrastructure...  ...foundation model training, large language model... 
    Training
    Full time
    Part time
    Local area

    Capital One

    New York, NY
    1 day ago
  • $290.4k - $363k

     ...reliable, production-grade systems that drive real...  .... Our agents are scalable systems built around...  ...the RoleAs a Senior/Staff Machine Learning Engineer (MLE) on the General...  ...Experience deploying ML systems in cloud environments...  ...education or training. Scale employees in... 
    Training
    Full time

    Scale AI

    New York, NY
    6 days ago
  • $232k - $258k

     ...at marketplace scale. In this Staff Machine Learning Engineering role, you will set technical direction...  ...and deliver end-to-end ML solutions that support distributed systems handling billions of trips....  ...scale, including development, training, productionization, and monitoring... 
    Training
    Full time
    Work at office
    Remote work

    Uber

    New York, NY
    13 hours ago
  • $400k

     ...Remote | Member of Technical Staff, Research Engineering — $400,000–$800,000/year...  ...in reinforcement learning, ML-oriented data systems, evaluation infrastructure, and scalable experimentation to contribute...  ...-learning environments, training pipelines, synthetic data systems... 
    Training
    Full time
    Remote work

    24-Mag Llc

    New York, NY
    14 hours ago
  • $295k

     ...business problems.We’re training and deploying...  ...are building AI systems. We believe that our...  ...of researchers, engineers, designers, and more...  ...-scale training, distributed systems, and HPC infrastructure...  ..., reliable, and scalable model training —...  ...the full stack of ML systems, this role... 
    Training
    Full time
    Work at office
    Local area
    Remote work
    Home office

    Cohere

    New York, NY
    6 days ago
  •  ...Across the Financial Connections Engineering org, we focus on delivering...  ...at scale — building the ML systems that transform raw financial...  ...play a key role in designing, training, and deploying ML models...  ...production systems and ensure their scalability and reliabilityCollaborate... 
    Training

    Stripe

    New York, NY
    2 days ago
  • $269.1k - $307.2k

    ## Staff AI EngineerApply: McLean, VA: San...  ...responsible and reliable AI systems, changing banking...  ...of AI & ML are bringing humanity...  ...science and engineering teams to deliver our...  ...product experiences and scalable, high-performance...  ...foundation model training, large language... 
    Training
    Full time
    Part time
    Local area

    Capital One

    New York, NY
    1 day ago
  •  ...team of registered architects, engineers, and city planners to...  ...AgentCore: Use Runtime for secure, scalable hosting and streaming;...  ...high-reliability backend or ML systems (Python/TypeScript), cloud infra...  ...ML‑ready data that shortens training cycles and lifts model... 
    Training
    Full time
    Summer work
    Work at office
    Flexible hours

    GreenLite

    New York, NY
    4 days ago
  • $269.1k - $307.2k

    Staff AI Engineer At Capital One, we are creating responsible and reliable AI systems, changing banking for good. For years, Capital...  ...of AI & ML are bringing humanity...  ...product experiences and scalable, high-performance AI...  ...including foundation model training, large language... 
    Training
    Full time
    Part time
    Local area

    Capital One

    New York, NY
    5 days ago
  • Fanatics is seeking a Staff Engineer for its Distributed Systems team within Fanatics Markets. You will own the event‑driven infrastructure powering real‑money trading, writing code with Kotlin/Java and Spring Boot, and shaping system design for peak load periods. You... 

    Fanatics

    New York, NY
    2 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff Engineer: Scalable ML Training & Distributed Systems. Be the first to apply!