Remote: Senior ML Systems Engineer — JAX/TPU at Scale
Fort Worth, Tarrant County, Texas, 76102, USA
Listed on 2026-10-09
-
Software Development
AssemblyAI is seeking a Senior Research Engineer to join the Research team, developing systems for large-scale distributed training, data processing, and inference. You’ll measure impact, own end-to-end experiments, and push for fast, reliable improvements across the pipeline.
The role requires deep knowledge of JAX/TPUs, model optimization, and production-grade inference. You’ll collaborate across research, infrastructure, and production engineering to deliver measurable value at scale.
We are seeking a motivated Remote:
Senior ML Systems Engineer — JAX/TPU at Scale to join AssemblyAI, Inc. in United States.
The Remote:
Senior ML Systems Engineer — JAX/TPU at Scale role at AssemblyAI, Inc. is now open for applications in United States.
Join us at AssemblyAI, Inc. as our next Remote:
Senior ML Systems Engineer — JAX/TPU at Scale in United States.
We are currently recruiting a Remote:
Senior ML Systems Engineer — JAX/TPU at Scale for our team in United States.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).