5 дней назад
ML Systems Engineer (AI)
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
ML Systems Engineer (AI): Building and optimizing distributed training and inference systems for large-scale foundation models with an accent on GPU performance, model parallelization, and memory efficiency. Focus on profiling compute and inter-GPU communication bottlenecks, developing Triton/CUDA kernels, and productionizing efficient model serving.
Location: Barcelona, on-site
Company
is an AI company developing NEXUS, a large tabular model for enterprise decision-making, founded by DeepMind alumni and backed by leading investors.
What you will do
- Profile end-to-end distributed training runs to identify compute, GPU memory, and inter-GPU communication bottlenecks.
- Contribute to architectural decisions that improve the efficiency and reliability of large-scale training jobs.
- Develop Triton and CUDA kernels when needed, and implement model scaling, parallelization, and memory optimization techniques for large-context workloads.
- Collaborate with ML Researchers to diagnose architectural inefficiencies and ensure research ideas scale efficiently in practice.
- Drive model productionization and serving, including inference optimization through techniques such as quantization.
Requirements
- Strong understanding of modern ML architectures and large-scale training pipelines.
- Experience running distributed training jobs on multi-GPU systems.
- Advanced profiling and debugging skills across CPU, GPU, memory usage, latency, and inter-GPU communication.
- Strong Python programming skills.
- Experience with model scaling and parallelization strategies, including tensor and pipeline parallelism.
- Familiarity with NCCL, MPI, PyTorch and Triton internals, plus programming experience with C++ and CUDA.
Culture & Benefits
- Competitive compensation with salary and equity.
- Comprehensive health coverage for employees and dependents.
- Paid parental leave for all new parents, including adoptive and surrogate journeys.
- Relocation support for employees moving to an office location.
- Low-ego, mission-driven culture focused on diversity of thought, ownership, and bias toward action.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →