Contentu
All jobs
A

Senior Research Engineer

AssemblyAI

WorldwideProgrammingLevel: Senior

Senior role on the Research team improving large-scale distributed training, data processing, and inference systems. Requires expert-level JAX and TPU skills, production inference optimization, and experience with LLMs or speech models. Also involves cross-functional work with research, infrastructure, and production engineering to resolve end-to-end bottlenecks.

jaxtpusflaxoptaxxladistributed traininginference optimizationpython

Location per source: Anywhere in the World

Apply on We Work Remotely

Source: We Work Remotely

Contentu does not verify employers. Never pay to apply for a job.