Research Engineer
Lightning AI6 days ago
London, United Kingdom +3 moreMid Level / Senior
Base Salary
$180k - $250k/yr
Responsibilities
- Develop performance-oriented model optimizations at graph, kernel, and system levels.
- Advance the Thunder compiler by building optimization passes and integration hooks.
- Ensure optimizations are accessible through clean APIs and automated tooling.
- Design and implement profiling and debugging tools to analyze model execution.
- Collaborate with hardware vendors to ensure efficient performance across diverse backends.
- Contribute to open-source projects by developing new features and improving documentation.
- Engage with the community to provide guidance on performance tuning.
Requirements
- Strong expertise with deep learning frameworks such as PyTorch.
- Hands-on experience with model optimization techniques.
- Knowledge of distributed systems and parallelism strategies.
- Familiarity with software engineering practices for performance-sensitive systems.
- Excellent collaboration and communication skills.
- Bachelor’s degree in Computer Science, Engineering, or a related field.
Benefits
- Comprehensive medical, dental, and vision coverage.
- Retirement and financial wellness support.
- Generous paid time off, plus holidays.
- Paid parental leave.
- Professional development support.
- Wellness and work-from-home stipends.
- Flexible work environment.