Software Engineer, Systems ML

In this role, you will design and build developer tools across the MTIA tooling ecosystem, with a primary focus on debugging, sanitizer technology, and fault isolation for AI workloads running on MTIA hardware at scale. You will work at the intersection of compilers, runtime, hardware, and ML frameworks, collaborating with cross-functional partners to deliver a high-quality developer experience for Meta's custom AI accelerators.

Responsibilities

Lead the design and development of MTIA's profiling and debugging tooling, including performance profiling, live debugging, and core dump for accelerator workloads Build tooling capabilities spanning performance profiling, kernel-level diagnostics, and multi-rank fault isolation Contribute broadly to the MTIA SW tooling infrastructure - profiling, performance debugging, memory analysis, monitoring, and reliability analysis for training and inference workloads Own significant components end-to-end, from technical design through implementation, production rollout, and long-term maintenance Collaborate closely with the MTIA compiler, runtime, kernel, and hardware teams to instrument the software stack with the hooks, sanitizers, and debuggers that are required Drive AI-native tooling approaches, leveraging automation and LLM-guided diagnostics to reduce time-to-root-cause Partner with internal product teams across advertising, recommendations, and generative AI to understand developer pain points and prioritize tooling investments Communicate architectural decisions through design documents and cross-team reviews; share debugging methodology and best practices with engineers across the MTIA stack

Qualifications

Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience Bachelor's degree in Computer Science, Computer Engineering, a relevant technical field, or equivalent practical experience 4+ years of experience in systems software engineering, performance engineering, developer tooling, or a closely related field Experience building debugging, sanitiser, profiling, or diagnostic tools for complex software/hardware systems Proficiency in C++ and Python, including low-level systems programming and scripting for tool automation Experience working across multiple layers of a system stack (compiler, runtime, OS/driver, hardware) Track record of leading the technical design and delivery of tooling or infrastructure projects through to production deployment Experience debugging issues across application, runtime, and hardware boundaries, with the ability to trace issues across application, runtime, and hardware boundaries Demonstrated ongoing AI skill development (e.g., prompt/context engineering, agent orchestration) and staying current with emerging AI technologies Experience adhering to and implementing responsible, ethical AI practices (e.g., risk assessment, bias mitigation, quality and accuracy reviews) Experience with accelerator ecosystems (GPU/CUDA, TPU, custom ASICs), including memory analysis, performance profiling, and runtime debugging using their toolchains (cuda-gdb, nsight-compute, nsight-systems) Familiarity with ML framework internals (PyTorch graph execution, torch.compile, operator dispatch) and AI compiler stacks (MLIR, LLVM, TVM, Triton) Demonstrated ability to integrate AI tools to optimize/redesign workflows and drive measurable impact (e.g., efficiency gains, quality improvements) Experience with Linux kernel and driver-level debugging, hands-on experience with sanitizer and dynamic-analysis technology, or building comparable memory checking tools Experience working at the device-software boundary - accelerator runtime/driver internals, DMA and memory-mapped device behavior, on-device fault handling or firmware-assisted error reporting Experience with Linux debugging and profiling infrastructure (gdb, perf, eBPF, ftrace, coredump analysis, hardware performance counters) and familiarity with binary formats and debugging metadata (ELF/DWARF) Experience with compiler-based instrumentation and runtime shadow-memory techniques underlying sanitizers (LLVM instrumentation passes, interceptors, allocator red zones) Track record of building developer tools adopted by large engineering populations, ideally with contributions to open-source projects (LLVM sanitizers, gdb, Valgrind, Triton, etc.) Bachelor's degree + 5+ years, Master's degree + 3+ years, or PhD + 2+ years in systems software, developer tooling, or accelerator software development