← All roles

Member of Technical Staff (GPU Performance Engineer)

Reka· Technical
Apply on company site
RemoteFull-timePosted 9mo ago

About the role

GPU Performance Engineer role focused on Python-based training infrastructure, large-scale model training, post-training processes (reinforcement learning, fine-tuning), and scalable model serving optimization.

Responsibilities

  • Design and implement improvements to training infrastructure and contribute to technical decisions optimizing model performance.
  • Work on post-training processes, including reinforcement learning and fine-tuning.
  • Contribute to improving efficiency and scalability of model serving infrastructure.

Qualifications

  • 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 (CUDA, C++).
  • Experience scaling up GPU jobs using large-scale compute clusters (e.g., Slurm or Kubernetes).
  • Demonstrated ability to analyze and optimize the performance of GPU-accelerated workloads, including profiling, identifying bottlenecks, and implementing performance tuning techniques.

Skills

Required:PythonPyTorchCUDA

Benefits

  • five weeks of paid leave
  • comprehensive healthcare benefits (including vision and dental)
  • additional perks that support well-being
  • Visa support including H1B and OPT transfers for US employees
Last updated Jul 16, 2026. Interested? Apply on the company site.

About Reka

Invention. Intelligence. Impact.

Invention. Intelligence. Impact. At Reka, we make the complex simple. Our multimodal, modular intelligence disassembles, understands, and delivers complete solutions in new configurations. Reka is intelligence fit for the real world.

AI solutionsartificial intelligencebessemer-ai-infrastructure-2026enterprise AIliquid-ai-recruiting-targetmodular AImultimodaltransformer models
Company profileWebsite

More open roles at Reka