Reka logoReka logo

GPU Performance Engineer - Member of Technical Staff

RekaUS, UK, Singapore, Remote
Remote Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Experience Level

Experience

Qualifications

Ideal ExperienceProficient in Python and PyTorch (or similar frameworks). Demonstrated experience in implementing and training extensive deep learning models. Strong background in writing and debugging low-level GPU code (CUDA, C++). Experience scaling GPU jobs on large compute clusters (e.g., Slurm, Kubernetes). Ability to analyze and optimize GPU-accelerated workloads, including profiling and performance tuning.

About the job

Join our dynamic team at Reka as a GPU Performance Engineer, where you will leverage your expertise in Python and large-scale model training to enhance our training infrastructure. You will play a pivotal role in optimizing model performance, contributing to critical technical decisions, and improving our post-training processes, including reinforcement learning and fine-tuning. Your contributions will also focus on enhancing the efficiency and scalability of our model serving infrastructure.

About Reka

At Reka, we are dedicated to developing impactful multimodal artificial intelligence solutions that empower organizations and businesses worldwide. As a globally distributed foundation model startup based in the San Francisco Bay Area, California, we embrace a remote-first culture, attracting top talent from across the globe. Our founding team and many of our members have played a significant role in advancing AI technology over the past decade.

Similar jobs

Browse all companies, explore by city & role, or SEO search pages.

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.