Назад
Company hidden
13 дней назад

GPU Optimization Engineer (AI)

Формат работы
remote (только Europe)
Тип работы
fulltime
Грейд
senior
Английский
b2
Страна
UK/Europe
Вакансия из списка Hirify.GlobalВакансия из Hirify Global, списка международных tech-компаний
Для мэтча и отклика нужен Plus

Мэтч & Сопровод

Для мэтча с этой вакансией нужен Plus

Описание вакансии

Текст:
/
TL;DR
GPU Optimization Engineer (AI): Optimizing multimodal model training and inference code for efficient large-scale deployment with an accent on CUDA, Triton, fused kernels, tensor cores, and distributed execution. Focus on profiling GPU and CPU workloads, designing custom PyTorch operations, and building monitoring automation that preserves performance gains.

Location: Remote within the EU

Company

hirify.global builds multimodal AI models designed to generate, understand, and operate in the physical world.

What you will do

  • Profile and optimize GPU, CPU, and accelerator code to maximize utilization and reduce latency.
  • Develop high-performance PyTorch, Triton, and CUDA kernels, including custom operations.
  • Build fused kernels using tensor cores and modern hardware features.
  • Optimize transformer model architectures and implementations for distributed multi-node production deployment.
  • Create performance monitoring, analysis, and automation tools.
  • Research and implement advanced optimization techniques for transformer models.

Requirements

  • Expert-level Triton and CUDA programming with GPU optimization experience.
  • Strong PyTorch skills, including kernel development and custom operations.
  • Proficiency with profiling tools such as NVIDIA Nsight, torch profiler, and custom tooling.
  • Deep understanding of transformer architectures and attention mechanisms.
  • Ability to work remotely from within the EU.

Nice to have

  • Experience with compilers and exporters such as torch.compile, TensorRT, ONNX, or XLA.
  • Experience optimizing inference workloads for latency and throughput.
  • Knowledge of Triton compiler technology and kernel fusion techniques.
  • Knowledge of warp-level intrinsics and advanced CUDA optimization.

Culture & Benefits

  • Full-time role in the Research & AI department.
  • Work remotely within the EU.
  • Equal opportunity employer.

Hiring process

  • The first 90 days focus on profiling current training and inference paths, shipping measurable kernel or architecture improvements, and building monitoring automation.

Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →