About the position
We are seeking an experienced engineer to work on distributed AI/ML systems. This role involves working on collective operations - the fundamental operations that enable AI to scale across multiple accelerators & servers. Most of our stack is C/C++ and relatively low level, so solid knowledge of Linux, kernels, and performant code is important. Experience with embedded systems is valued, and experience with high-speed networking or HPC interconnects is valued highly. If you like solving hard problems, want to work with HPC and ML customers, iterate fast and deliver meaningful solutions at scale, then come join us! This truly is a role on the forefront of AI/ML, you'll be working on features for the largest clusters, with the largest customers, for the largest AI models.
Responsibilities
- Work on collective operations that enable AI to scale across multiple accelerators and servers.
- Develop hardware and software components for EC2 infrastructure.
- Build networking solutions for Machine Learning (ML) and High-Performance Computing (HPC) workloads on AWS.
- Collaborate with infrastructure experts, hardware engineers, RTL engineers, scientists, and architects.
- Mentor new and junior engineers while receiving mentorship from senior members.
- Participate in code reviews and knowledge-sharing sessions.
Requirements
- 3+ years of non-internship professional software development experience.
- 2+ years of non-internship design or architecture experience of new and existing systems.
- Experience programming with at least one software programming language.
Nice-to-haves
- 3+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
- Bachelor's degree in computer science or equivalent.
Benefits
- Flexible working hours.
- Work-life balance support.
- Mentorship and career growth opportunities.
- Inclusive team culture.
