Latest AI JobsBrowse Jobs
Reka logo

Member of Technical Staff (GPU Performance Engineer)

RekaUS, UK, Singapore, RemoteTechnical
Apply on company site
Salary
Not listed
Job type
Full-time
Work mode
Remote
Posted
8 Jan 2026
Status
Open · checked 7d ago

About this role

Role Overview

The Member of Technical Staff (GPU Performance Engineer) position at Reka is an exciting opportunity for an experienced engineer to optimize large-scale model training infrastructure remotely or across the US, UK, and Singapore. This role suits a technical professional with a deep background in Python, PyTorch, and low-level GPU code who wants to advance multimodal artificial intelligence.

Key Responsibilities

Design and implement enhancements to training infrastructure to directly impact model performance.
Contribute to key technical decisions aimed at maximizing the efficiency of deep learning models.
Engage in post-training processes such as fine-tuning and reinforcement learning.
Enhance the scalability and overall efficiency of the model serving infrastructure.

Requirements

Strong engineering abilities featuring fluency in Python and PyTorch or related frameworks.
Demonstrated background in implementing and training large deep learning models.
Experience writing and debugging low-level GPU code including C++ and CUDA.
Track record of scaling up GPU jobs with large-scale compute clusters like Kubernetes or Slurm.
Proven skill in analyzing and optimizing GPU-accelerated workloads through profiling, identifying bottlenecks, and applying performance tuning techniques.

Skills and Tools

Python
PyTorch
Kubernetes
Slurm
C++
CUDA

Salary and Benefits

Comprehensive healthcare coverage including dental and vision benefits.
Five weeks of paid leave.
Visa assistance including H1B and OPT transfers for US-based employees.
Collaborative work environment with a globally distributed, remote-first team.

About the Company

Reka is a globally distributed foundation model startup headquartered in the San Francisco Bay Area in California. Operating with a remote-first mindset, the organization brings together top-tier talent from around the world. The founding team and many staff members have delivered numerous artificial intelligence breakthroughs over the past decade, driven by a mission to build useful multimodal AI that empowers businesses and organizations.

How to Apply

Interested candidates should visit the official Reka careers page to submit their application for this remote position.

Skills and tech stack

PythonPyTorchKubernetesC++

Questions candidates ask

What is the location for this role?

The position offers remote flexibility across the US, UK, and Singapore, and the company operates as a globally distributed, remote-first startup headquartered in the San Francisco Bay Area.

What programming languages and frameworks are required?

Candidates need strong engineering skills with fluency in Python and PyTorch or other frameworks, alongside experience writing and debugging low-level GPU code such as C++ and CUDA.

What benefits does Reka offer?

Employees receive five weeks of paid leave, comprehensive healthcare benefits including vision and dental, and visa assistance such as H1B and OPT transfers for US employees.

What are the main responsibilities of the GPU Performance Engineer?

The engineer will design and implement training infrastructure improvements, make technical decisions to optimize model performance, work on reinforcement learning and fine-tuning, and improve model serving efficiency and scalability.

How should I apply for the position?

Candidates should apply directly on the official Reka company careers page.

Ready to apply?

The application is on the official Reka careers page.

Apply on company site