Member of Technical Staff (GPU Performance Engineer)

Member of Technical Staff (GPU Performance Engineer) in Singapore, UK, USA

Reka is seeking a senior-level Member of Technical Staff (GPU Performance Engineer) to join our globally distributed, remote-first team. In this engineering role, you will design and implement training infrastructure improvements, optimize large-scale model performance, and work on post-training processes like reinforcement learning and fine-tuning. You will also contribute to making our model serving infrastructure more efficient and scalable.

Core Requirements

• Bachelors degree and 5+ years of relevant professional experience
• Strong engineering skills with fluency in Python and PyTorch or other frameworks
• Proven experience implementing and training large deep learning models
• Experience writing and debugging low-level GPU code including CUDA and C++
• Experience scaling up GPU jobs using large-scale compute clusters such as Slurm or Kubernetes
• Demonstrated ability to analyze and optimize GPU-accelerated workloads through profiling, bottleneck identification, and performance tuning

What We Offer

As part of our remote-first team, you will collaborate with top-tier talent from renowned AI organizations. We offer cutting-edge infrastructure, an inclusive culture, five weeks of paid leave, comprehensive healthcare benefits covering vision and dental, and visa assistance including H1B and OPT transfers for US employees. Salary is market competitive.