4 часа назад
Staff Software Engineer (SIMD Kernels)
190 000 - 300 000$
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
Staff Software Engineer (SIMD Kernels) (AI hardware/software): Developing and maintaining SIMD software kernels for ML operators such as softmax, layer normalization, and activation functions for next-generation AI hardware with an accent on algorithm mapping, hardware–software co-design, and SDK usability. Focus on implementing and scaling C/C++ and Python kernels across specialized hardware architectures, analyzing performance, and supporting ML compiler and framework integration.
Location: Santa Clara, California headquarters, regional offices, or remote within the United States; hybrid work is available.
Salary: $190,000–$300,000 per year, plus equity and bonus opportunities.
Company
develops software and hardware for generative AI and AI compute systems.
What you will do
- Develop, enhance, and maintain SIMD software kernels for ML operators including softmax, layer normalization, activation functions, GEMMs, convolutions, and pooling.
- Productize the software stack for an AI compute engine and next-generation AI hardware.
- Map algorithms and AI framework computational graphs onto modern hardware architectures.
- Build SDK solutions that improve developer usability and enable performance analysis.
- Navigate hardware–software co-design trade-offs and deliver scalable, high-quality software.
Requirements
- MS or PhD in computer engineering, mathematics, physics, or a related field, plus 5+ years of industry experience.
- Strong knowledge of computer architecture, data structures, system software, and machine learning fundamentals.
- Proficiency in C/C++ and Python development in a Linux environment.
- Experience implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs, or AI accelerators, including CUDA-based development.
- Experience implementing operators used in ML workloads and a strong sense of ownership and leadership.
Nice to have
- Startup, small-team, or incubation experience.
- Experience with TensorFlow, PyTorch, MLIR, LLVM, TVM, or Glow.
- Experience with deep learning models for computer vision, NLP, or recommendation systems.
- Experience developing for embedded SIMD vector processors such as Tensilica.
- Experience at a cloud provider or AI compute/subsystem company.
Culture & Benefits
- Collaborative and inclusive environment centered on respect, humility, and direct communication.
- Medical, dental, and vision insurance.
- 401(k) and an employee rewards plan focused on wellbeing.
- Equity and performance-based bonus opportunities.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →
Похожие вакансии
5 дней назад
AI Infrastructure Engineer (GPU)
170 500 - 315 490$
5 часов назад
Distributed LLM Inference Engineer (AI)
170 000 - 245 000$
2 дня назад
AI Developer (Applied ML Practitioner)
155 000 - 162 500$
Microsoft AI
7 дней назад
Member of Technical Staff, Compute Orchestration & Scheduling (AI)
188 000 - 304 200$
1 день назад
Software Engineer, Applied AI (AI)
150 000 - 170 000$
1 день назад
AI Software Engineer
150 000 - 200 000$