AssemblyAI

Research Engineer

• Investigate and mitigate performance bottlenecks in large-scale distributed training and inference systems.
• Develop and implement low-level (operator/kernel) and high-level (system/architecture) optimization strategies.
• Translate research models and prototypes into highly optimized, production-ready inference systems.
• Explore and integrate inference compilers such as TensorRT, ONNX Runtime, AWS Neuron and Inferentia.
• Design, test, and deploy scalable solutions for parallel and distributed workloads on heterogeneous hardware.
• Facilitate knowledge transfer and bidirectional support between Research and Engineering teams, ensuring alignment of priorities and solutions.
• Collaborate closely with Research and Engineering teams to bridge research and production engineering.

Senior Research Engineer

Platform Engineer

• Drive technical direction of data and AI training infrastructure team and shape vision and architecture
• Enhance and scale infrastructure to improve training velocity and research outcomes
• Design scalable, future-proof data platforms optimized for AI research workloads
• Build efficient data pipelines leveraging GCP services
• Implement cost-effective storage and monitoring solutions for ML at scale
• Create flexible training resource management with intelligent queuing and optimize resource allocation
• Participate in on-call rotation to ensure system reliability
• Lead adoption of cutting-edge ML tools and frameworks and streamline workflows
• Enhance tooling and documentation, implement guardrails for cost, quality, and performance
• Collaborate closely with Research team and senior leadership and provide technical guidance across the organization

Senior Research Engineer

Lead AI Platform Engineer