Senior Research Engineer: JAX/TPU Scale & Inference
Listed on 2026-10-31
-
Software Development
AssemblyAI is seeking a Senior Research Engineer to advance large-scale distributed training, data processing, and inference for voice AI models. You will push experimental velocity, refine the JAX/TPU framework, and join cross-functional teams to ship optimized, production-ready systems.
The role emphasizes deep systems expertise, measurement discipline, and hands-on work across training, evaluation, and deployment with a focus on low-latency inference and scalable architectures.
This is a great opening to take on the Senior Research Engineer: JAX/TPU Scale & Inference role at Remote.
Full responsibilities and requirements are described in the listing above.
Learn more about the Senior Research Engineer: JAX/TPU Scale & Inference role in the description above.
We appreciate your interest in this position.
Join Remote and contribute to our ongoing work.
Take a moment to read everything above and see whether this role is right for you.
(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).