Senior Applied Research Engineer | AI & ML | Spain, Barcelona
5 days ago
Barcelona
Senior Applied Research Engineer | Barcelona, Spain Si sus habilidades, experiencia y cualificaciones coinciden con las de esta descripción del puesto, no demore su solicitud. We are partnered with a cutting-edge AI company shaping the future of enterprise decision-making. Founded by experienced technologists from leading research environments, the firm has developed a market-leading platform purpose-built for the structured data that underpins critical business decisions. Backed by top-tier investors and trusted by some of the world’s largest organisations, the company helps enterprises unlock significant value by enabling more accurate, forward-looking decision-making. You will work on novel technical challenges in large-scale model development and contribute to technology that is changing how major organisations operate. This is an opportunity to join a category-defining company at an early stage and help shape its trajectory. Barcelona, Spain, Hybrid working Competitive Salary + Equity + Benefits Permanent Role Key responsibilities • Profile end-to-end distributed training runs to identify bottlenecks across compute, GPU memory, and inter-GPU communication., • Influence architectural decisions to improve efficiency and reliability of large-scale training jobs, including developing Triton/CUDA kernels when needed., • Design and implement model scaling, parallelisation, and memory optimisation techniques for training workloads with very large context sizes., • Collaborate closely with ML Researchers to diagnose architectural inefficiencies, ensure new research ideas scale efficiently in practice, and share internal knowledge on optimisation., • Strong understanding of modern ML architectures and large-scale training pipelines., • Hands-on experience running distributed training jobs on multi-GPU systems., • Advanced profiling and debugging across CPU, GPU, memory usage, latency, and inter-GPU communication., • Strong programming skills in Python., • Familiarity with NCCL, MPI, and distributed communication primitives., • Knowledge of PyTorch and Triton internals., • Programming experience with C++ and CUDA. Benefits • Competitive compensation with salary and equity and comprehensive benefits, • Relocation support for employees moving to join the team in an office location. xcskxlj, • A mission-driven, low-ego culture valuing diversity of thought, ownership, and bias towards action. If you are interested in this role, please respond directly to this advert with your updated CV or email it to