AgentJobsPost a job

ML Engineer, Inference & Optimization

Mid
Apply now

About the role

ABOUT THE ROLE We are seeking Senior/Staff level Inference Engineers to accelerate the performance of Pika's AI-driven products. In this highly technical role, you will operate at the intersection of cutting-edge inference acceleration, GPU parallelism, advanced model deployment, and video generation technologies. Your expertise will drive significant improvements to model speed and efficiency, ensuring our creative AI systems deliver industry-leading user experiences at scale.   You will design and optimize inference pipelines, implement state-of-the-art acceleration techniques, and work closely with researchers and engineers across the team to push the boundaries of what’s possible in real-time AI deployment. Your efforts will play a foundational role in powering the next generation of Pika’s video and language models.   WHAT YOU’LL DO - Accelerate Inference: Lead and implement advanced inference acceleration techniques, including attention optimization and quantization for efficient model serving. - Maximize GPU Parallelism: Engineer and optimize GPU strategies across tensor, sequence, and pipeline parallelism (TP, SP, PP) for maximal efficiency and scalability. - Programming for Performance: Develop and optimize high-performance computing kernels and distributed workloads using CUDA and NCCL. - Advance AI Deployment: Collaborate with research and engineering teams to bring state-of-the-art videogen and large language models into production. - Improve Train…

Read the full description on the company site

Related jobs