Reka AI
Member of Technical Staff (GPU Performance Engineer)
US, UK, Singapore, Remote · Remote · Staff · Full-time · Posted Jan 8, 2026
- Visa sponsorship
- Visa assistance, including H1B and OPT transfers, for US employees.
About the job
Designs and implements improvements to large-scale model training and serving infrastructure, optimizing GPU performance and contributing to post-training processes.
What you'll do
- Improve training infrastructure and optimize model performance.
- Work on reinforcement learning and fine-tuning post-training processes.
- Improve the efficiency and scalability of model serving infrastructure.
What they ask for
- Fluency in Python and PyTorch or other frameworks.
- Experience implementing and training large deep learning models.
- Experience writing and debugging low-level GPU code (CUDA, C++).
- Experience scaling GPU jobs on large compute clusters and profiling and optimizing GPU workloads.
Tools
Python, PyTorch, CUDA, C++, Slurm, Kubernetes
AI focus
Work on large-scale model training, reinforcement learning, fine-tuning, and model serving infrastructure.
Model-written summary · the original posting is authoritative