Senior Research Engineer - JAX, TPU & Distributed Training
Job in
Northern, Floyd County, Kentucky, USA
Listed on 2026-09-08
Listing for:
Neura Market
Apprenticeship/Internship
position Listed on 2026-09-08
Job specializations:
-
Software Development
Job Description & How to Apply Below
AssemblyAI is seeking a Senior Research Engineer to advance large-scale distributed training, data processing, and inference. You will optimize JAX/TPU pipelines, build evaluation harnesses, and collaborate across researchers, infrastructure, and production engineering to ship robust, scalable AI systems.
You will work on end-to-end model development, measurement-driven improvements, and production-grade deployments, focusing on SLAs, latency, and accuracy for speech-related models.
#J-18808-LjbffrPosition Requirements
10+ Years
work experience
To View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(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).
(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).
Search for further Jobs Here:
×