Software Engineer, Systems Machine Learning

What you'll do

• Design, build, test, and deploy tools across the MTIA tooling ecosystem, including debugging, profiling, performance and memory analysis, and monitoring.
• Improve tooling infrastructure and integrate it with MTIA compiler, runtime, kernel, and hardware capabilities.
• Work with internal product teams to identify developer needs and improve tool usability and productivity.

What they ask for

• Bachelor’s degree in Computer Science, Computer Engineering, a relevant technical field, or equivalent practical experience; degree must be completed before joining.
• 2+ years of software engineering experience, with exposure to systems software, developer tooling, or infrastructure.
• Proficiency in C++ and Python, including systems-level programming concepts.
• Experience working across system-stack layers and independently delivering software from design through production deployment.

Nice to have

• Experience with accelerator ecosystems or heterogeneous computing; developer tools; Linux debugging and profiling tools; ML frameworks or compiler infrastructure; or distributed systems debugging and monitoring.
• Ongoing AI skill development, responsible AI practices, or demonstrated use of AI tools to improve workflows.

Pay

$121,992/year to $181,000/year (base salary)

Tools

C++, Python, GPU/CUDA, TPU, gdb, perf, strace, eBPF

AI focus

Builds tooling for AI workloads on MTIA hardware and partners with generative AI teams. The posting also seeks ongoing AI skill development and the ability to use AI tools to improve workflows.