Member of Technical Staff (GPU Engineer) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Reka AI · 2 days ago

Member of Technical Staff (GPU Engineer)

positionUnited States
timeFull-time
remoteRemote
seniorityMid, Senior Level

Reka AI is a globally distributed foundation model startup focused on building useful multimodal artificial intelligence. They are seeking an experienced GPU Engineer to design and implement improvements to their training infrastructure, optimize model performance, and contribute to post-training processes including reinforcement learning and fine-tuning.

Artificial Intelligence (AI)Foundational AIGenerative AIInformation TechnologySoftware
check
H1B Sponsorednote

Responsibilities

Design and implement improvements to our training infrastructure
Contribute to technical decisions that optimize performance of our models
Work on post-training processes, including reinforcement learning and fine-tuning
Contribute to improving the efficiency and scalability of our model serving infrastructure

Qualification

PythonPyTorchCUDALarge-scale model trainingC++GPU performance optimizationSlurmKubernetes

Required

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

Benefits

Five weeks of paid leave
Comprehensive healthcare benefits (including vision and dental)
Additional perks that support your well-being
Visa Support

Company

Reka AI

twittertwittertwitter
company-logo
Reka AI is an AI research and product company that develops multimodal AI solutions to advance science and build generative AI models.
dateFounded in 2022
location
Sunnyvale, California, USA
people-size11-50 employees
web-link
https://reka.ai

H1B Sponsorship

Reka AI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2)
2023 (1)

Funding

Current Stage
Growth Stage
Total Funding
$170M
2025-07-22Series B· $110M
2023-06-27Series A· $60M

Leadership Team

M
Mikel Artetxe
Co-Founder
linkedin

Recent News

PR Newswire
Oracle Unveils New Defense Ecosystem Members
2025-10-13
Carahsoft Technology Corporation
Reka and Carahsoft Partner to Deliver Custom AI Models for Government Agencies
2025-08-29
Crowdfund Insider
AI and Fintech Fueled $100M+ Mega-Rounds in July : Research
2025-08-18
Company data provided by crunchbase