company_site

Software Engineer, MTIA SW Performance Autotuning

Job Fit Check

Base Career helps you apply smarter for this job.

Ready to Scan

Key skills for this role

PythonC++PyTorch

Own the performance autotuning strategy for MTIA inference and training — current and future chip generations
Build and maintain our autotuning infrastructure: search spaces, cost models, benchmarking harnesses, and performance regression detection
Define how we trade off search cost against performance gains, and where autotuning fits in the compile/deploy pipeline
Partner with kernel and compiler teams to expose tunable parameters and encode them into an automated search
Debug performance issues that span multiple layers — kernel, runtime, compiler, or serving infra — and drive them to root-cause
Collaborate with product teams to accelerate model performance on MTIA
Set technical direction and help the team grow. We've scaled fast and need people who can elevate the impact around them, not just ship their own code

Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
Strong Python and C++ skills, with hands-on experience across the PyTorch stack
4+ years in ML systems, AI infra, performance engineering, or similar
Experience driving problems that span multiple teams, where no one owns the whole picture
Experience with accelerator performance concepts — roofline analysis, memory bandwidth, occupancy, and what makes kernels fast or slow on hardware
Track record of setting technical direction and mentoring engineers
BS in CS, CE, Math, or equivalent experience

Experience adhering to and implementing responsible, ethical AI practices (e.g., risk assessment, bias mitigation, quality and accuracy reviews)

Skip the repetitive application forms

Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.

Trusted by over 500,000 job seekers on Base Career