Research Engineer, AI Systems & Performance Optimization

Lightning AI; Town of Tonawanda (NY); Hybrid; USD 120,000 - 250,000; Full time

Benefits offered by this job

Comprehensive medical, dental and vision coverage

Generous paid time off, plus holidays

Professional development support

Job summary

Lightning AI is seeking a Research Engineer in New York who will optimize AI training and inference workloads. The role requires expertise in deep learning frameworks, particularly PyTorch, and offers the opportunity to work with cross-functional teams. Ideal candidates should have strong software engineering fundamentals and experience with large-scale AI systems.

Enjoy a competitive base salary and a comprehensive benefits package including medical coverage, retirement support, and generous paid time off.

Qualifications

• Strong expertise with PyTorch is essential.
• Experience in large-scale AI workloads is required.
• Familiarity with parallelism strategies is preferred.

Responsibilities

• Optimize large-scale training and inference workloads across GPUs.
• Work directly with customers to analyze workloads and improve performance.
• Develop tools for production AI workloads.

Skills

Deep learning frameworks (PyTorch)

Large-scale training or inference workloads

Distributed systems familiarity

Strong software engineering fundamentals

Performance bottlenecks analysis

Collaboration and communication skills

Ability to work in ambiguous environments

Education

Bachelor's degree in Computer Science, Engineering or related field

Tools

CUDA

TensorRT

Triton

Job description

Lightning AI is seeking a Research Engineer in New York who will optimize AI training and inference workloads. The role requires expertise in deep learning frameworks, particularly PyTorch, and offers the opportunity to work with cross-functional teams. Ideal candidates should have strong software engineering fundamentals and experience with large-scale AI systems.

Enjoy a competitive base salary and a comprehensive benefits package including medical coverage, retirement support, and generous paid time off.