AI Performance Optimization Engineer
Benefits:
Medical, dental and vision
Life and AD&D insurance
Flexible paid time off
Paid family leave benefits
$500 monthly meal reimbursement
Education Requirements:
Bachelor’s degree in Computer Science, Engineering, or a related field
Advanced degree (Master’s or PhD) in machine learning, compilers, or systems highly preferred
Experience Requirements:
Strong expertise with deep learning frameworks (PyTorch, JAX, or TensorFlow)
Experience with model optimization techniques (quantization, pruning)
Deep understanding of compiler internals
Experience with CUDA, Triton, or GPU programming
Knowledge of distributed systems and parallelism strategies
Other Requirements:
Track record contributing to open-source projects
Familiarity with software engineering practices (APIs, testing, CI/CD)
Excellent collaboration and communication skills
Responsibilities:
Develop performance-oriented model optimizations at graph, kernel, and system levels
Advance the Thunder compiler by building optimization passes
Work across software stack for clean APIs and automated tooling
Design and implement profiling and debugging tools
Collaborate with hardware vendors and ecosystem partners
Show more details