Tech Stack

Description

You will optimize training and inference workloads on compute accelerators and clusters using the Lightning Thunder compiler and PyTorch Lightning ecosystem, working at the intersection of deep learning research, compiler development, and large-scale system optimization.

Requirements

• Strong expertise with deep learning frameworks such as PyTorch
• Hands-on experience with model optimization (graph-level, quantization, pruning, mixed precision)
• Knowledge of distributed systems and parallelism strategies
• Familiarity with software engineering practices (APIs, tooling, CI/CD)
• Bachelor's degree in Computer Science or Engineering

Responsibilities

• Develop performance-oriented model optimizations at graph-level, kernel-level, and system-level
• Advance the Thunder compiler with optimization passes and graph transformations
• Design profiling and debugging tools to analyze model execution
• Collaborate with hardware vendors for efficient multi-backend support
• Contribute to open-source projects and community engagement