5 часов назад
Senior Staff Software Engineer - Kernels (AI)
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
Senior Staff Software Engineer - Kernels (AI) (C/C++/Python): Developing and maintaining software kernels for a next-generation AI compute engine with an accent on hardware-software co-design, algorithm mapping, and ML workloads. Focus on optimizing computational graphs and operators for specialized hardware, building compiler infrastructure with compiler experts, and scaling software deliverables in tight development windows.
Location: Hybrid, onsite at the Belgrade, Serbia office 3–5 days per week
Company
develops software and hardware for generative AI and next-generation AI compute engines.
What you will do
- Develop, enhance, and maintain software kernels for next-generation AI hardware.
- Productize the software stack for an AI compute engine.
- Implement and optimize algorithms for specialized hardware architectures, including FPGAs, DSPs, GPUs, and AI accelerators.
- Map computational graphs from AI frameworks to underlying hardware architectures.
- Work with compiler experts to build compiler infrastructure.
- Collaborate with ML, systems, mixed-signal, DSP, and CPU engineering specialists on hardware-software co-design.
Requirements
- MS in computer engineering, mathematics, physics, or a related field with 10+ years of industry experience, or a PhD in a related field with 1+ years of industry experience.
- Strong understanding of computer architecture, data structures, system software, and machine learning fundamentals.
- Proficiency in C/C++ and Python development in Linux environments.
- Experience implementing ML operators such as GEMMs, convolutions, BLAS, SIMD operations, softmax, layer normalization, and pooling.
- Experience developing for embedded SIMD vector processors such as Tensilica.
- Self-motivated approach, ownership, leadership, and collaborative teamwork.
Nice to have
- Startup, small-team, or incubation experience.
- Experience with TensorFlow, PyTorch, or other deep learning frameworks.
- Experience with ML compilers and algorithms, including MLIR, LLVM, TVM, or Glow.
- Experience with machine learning models for computer vision, NLP, or recommendation systems.
- Experience at a cloud provider or AI compute/subsystem company.
Culture & Benefits
- Collaborative and inclusive environment built around respect, humility, and direct communication.
- Focus on execution, ownership, continuous learning, and tackling technically challenging problems.
- Equal opportunity workplace committed to a welcoming and empowered workforce.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →