SW Team Lead
AragoВ офисеПолная занятость
PythonAIC++MLCGPUFirmwareEmbedded SystemsKernelCPUDriversSystem Software
Описание
As a Software Team Lead, you'll coordinate the efforts of the software team across the stack, driving performance optimization, deploying pilot and prototype stacks to grow the developer base, and ensuring alignment between client specifications and deliverables. You'll build and mentor a growing team while staying hands-on with code, architecture, and performance yourself, working closely with chip architects and hardware teams during early silicon bring-up.
Responsibilities
- Own and drive the software stack for AI accelerator products, balancing speed of execution with long-term technical foundations
- 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 as needed
- 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, translate them into clear team deliverables, and align scope, priorities, and timelines
Requirements
- 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, including translating client requirements into technical scope, deliverables, and realistic timelines
- Strong background in testing and deployment infrastructure for large-scale software projects
- Language: English at a proficient level
Nice to have
- Experience with Python and modern ML frameworks (PyTorch, ONNX, TensorRT, etc.) is highly desirable
- 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
We offer
- Competitive cash compensation, with final package based on location, experience, and the pay of team members in similar positions
- Meaningful stock option plan offered (included in the majority of full time offers)
- Healthcare coverage (including family-friendly options), pension contributions, professional development support, and 25 days of PTO, in addition to public holidays
- Ownership of a key technical domain, with significant vertical and/or horizontal growth opportunities, based on performance and individual drive