Hybrid ML Compiler & Performance Engineer

Voltage Park; New York (NY); Hybrid; USD 180,000 - 250,000; Full time

Benefits offered by this job

Medical, dental, and vision coverage

Flexible paid time off

$500 monthly meal reimbursement

$1,000 annual learning & development stipend

Job summary

Voltage Park is seeking a Research Engineer in New York City to optimize deep learning workloads using the Lightning Thunder compiler. This hybrid role involves developing model optimizations, collaborating with hardware vendors, and integrating improvements with PyTorch Lightning.

The desired candidate will have strong expertise in deep learning frameworks, model optimization techniques, and excellent collaboration skills. Benefits include a competitive salary, medical coverage, and additional perks such as meal reimbursements and learning stipends.

Qualifications

• Strong expertise with deep learning frameworks such as PyTorch.
• Hands-on experience with model optimization techniques like graph-level optimizations.
• Knowledge of distributed systems and parallelism strategies.

Responsibilities

• Develop performance-oriented model optimizations at multiple levels.
• Advance the Thunder compiler by building optimization passes and integration hooks.
• Work across the software stack to expose optimizations via clean APIs.

Skills

Deep learning frameworks (PyTorch)

Model optimization techniques

Distributed systems knowledge

Software engineering practices

Collaboration and communication

Education

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

Tools

CUDA

Triton

Job description

Voltage Park is seeking a Research Engineer in New York City to optimize deep learning workloads using the Lightning Thunder compiler. This hybrid role involves developing model optimizations, collaborating with hardware vendors, and integrating improvements with PyTorch Lightning.

The desired candidate will have strong expertise in deep learning frameworks, model optimization techniques, and excellent collaboration skills. Benefits include a competitive salary, medical coverage, and additional perks such as meal reimbursements and learning stipends.