Research Engineer

Role Overview

A highly skilled Research Engineer is needed to optimize large-scale training and inference workloads on AI infrastructure.

Key Responsibilities

• Enhance performance, scalability, and reliability of AI systems across GPUs, accelerators, and distributed platforms.
• Collaborate with customers to analyze workloads, identify bottlenecks, and implement improvements.
• Develop inference pipelines, model serving systems, and performance tooling for production.
• Design profiling, debugging, and observability tools to guide optimization.
• Ensure performance enhancements are integrated via APIs, automation, and the Lightning ecosystem.
• Partner with hardware vendors to support diverse compute backends and contribute to open-source projects.

Qualifications & Skills

• Expertise in deep learning frameworks like PyTorch.
• Experience with large-scale training or inference workloads and distributed systems.
• Strong software engineering skills, API design, debugging, and production deployment.
• Excellent collaboration and communication abilities.
• Bachelor's in Computer Science, Engineering, or related field; advanced degrees preferred.