Strong expertise with pytorch deep learning frameworks
Significant experience with large-scale training or inference workloads
Deep understanding of distributed systems and parallelism strategies
Lightning AI combines developer-first software with cost-efficient, large-scale compute to take ideas from research to production
Job Summary
Lightning AI combines developer-first software with cost-efficient, large-scale compute to take ideas from research to production.
This role sits at the intersection of ML systems, AI infrastructure, performance engineering, and practical research to drive improvements across models and platforms.
The company offers a comprehensive benefits package including medical coverage, retirement support, and flexible work environments.
Matching Summary
Lightning AI combines developer-first software with cost-efficient, large-scale compute to take ideas from research to production.
Salary
Base: $225,000 - $275,000 USD annually; Bonus/Equity: Discretionary bonus and meaningful equity component included; Benefits: Comprehensive medical, dental, vision, pension, and paid time off
Skills & Requirements
Must-have
Strong expertise with PyTorch deep learning frameworks
Significant experience with large-scale training or inference workloads
Deep understanding of distributed systems and parallelism strategies
Hands-on experience with inference optimization techniques
Experience with modern ML systems tooling like TensorRT or vLLM
Nice-to-have
Experience contributing to open-source ML infrastructure projects
Experience working closely with hardware vendors or accelerator ecosystems
Startup experience in highly cross-functional environments
Experience mentoring engineers or leading technical initiatives
Key Requirements
Master's or PhD in Computer Science, AI, or related field
Proven track record leading performance optimization efforts
Ability to operate effectively in ambiguous, fast-moving environments