SW Team Lead

The role

You will coordinate software team efforts across the stack and drive performance optimization. You will build and mentor a team while staying hands-on with code, architecture, and performance. You will work with chip architects and hardware teams during early silicon bring-up.

What you will do

• Own and drive the software stack for AI accelerator products.
• Work closely with chip architecture and hardware teams on hardware–software co-design and early silicon bring-up.
• Lead development across kernel libraries, runtimes, drivers, firmware, and developer tooling.
• Optimize key AI workloads for performance, efficiency, memory usage, and reliability on emerging hardware.
• Stay hands-on with architecture, coding, debugging, reviews, and performance analysis while mentoring a growing team.
• Partner directly with clients to understand technical requirements and translate them into deliverables.

Required qualifications

• Strong background in systems software for AI/ML accelerators, GPUs, CPUs, or other high-performance compute platforms.
• Solid understanding of computer architecture, memory systems, parallel computing, and hardware-software interaction.
• Proven ability to lead small engineering teams while retaining significant individual contribution.
• Strong C/C++ programming skills.
• Strong communication and project ownership skills.
• Strong background in testing and deployment infrastructure for large-scale software projects.
• English at a proficient level.

Nice to have

• Experience with Python and modern ML frameworks (PyTorch, ONNX, TensorRT, etc.)
• Strong background in mathematics.
• Hands-on experience with low-level software such as compilers, drivers, and firmware.
• Experience profiling and optimizing AI workloads for performance, memory efficiency, and hardware utilization.
• Experience with AI inference platforms such as vLLM and SGLang.

Benefits

• Competitive cash compensation
• Meaningful stock option plan
• Healthcare coverage (including family-friendly options)
• Pension contributions
• Professional development support
• 25 days of PTO, in addition to public holidays

About the company

Arago's mission is to re-engineer the foundations of computing from first principles. The company uses proprietary technology that fuses optical and CMOS technologies to deliver an order-of-magnitude increase in performance.