Member of Technical Staff (GPU Performance Engineer)

Reka
United States, United Kingdom, Singapore
Workplace: RemoteFull timeFunction: Software EngineeringSkills: ["Problem-solving","Communication","Teamwork"]

Seeking an experienced GPU Performance Engineer with strong Python and large-scale model training background to design and implement improvements to training infrastructure, influencing technical decisions to boost model performance. You will also work on post-training processes including reinforcement learning and fine-tuning, and help improve the efficiency and scalability of the model serving infrastructure.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Reka
Reka
7 months ago

Member of Technical Staff (GPU Performance Engineer)

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 22 hours agoStatus: Live

Job Summary

Seeking an experienced GPU Performance Engineer with strong Python and large-scale model training background to design and implement improvements to training infrastructure, influencing technical decisions to boost model performance. You will also work on post-training processes including reinforcement learning and fine-tuning, and help improve the efficiency and scalability of the model serving infrastructure.
Location: United States, United Kingdom, Singapore
Workplace: Remote
Employment Type: Full time
Job Function: Software Engineering

Key Responsibilities

  • •Design and implement improvements to training infrastructure to optimize model performance.
  • •Influence technical decisions related to model training and deployment for performance gains.
  • •Contribute to post-training processes, including reinforcement learning and fine-tuning.
  • •Improve efficiency and scalability of model serving infrastructure.
  • •Collaborate with top engineers and researchers to advance AI infrastructure.

Pay and Benefits

Perks:Paid LeaveHealth InsuranceVisionDental

Key Requirements

  • •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).
  • •Ability to analyze and optimize the performance of GPU-accelerated workloads, including profiling and bottleneck identification.
Experience:AIMachine learningGPU computingDeep learning
Skills:Problem-solvingCommunicationTeamwork
Certifications:H1BOPT
Tech Stack:PythonPyTorchCUDAC++SlurmKubernetes

Eligibility

Visa:H1BOPT

Company Brief

Reka
Develops AI-driven software and tooling to accelerate development, testing, and deployment of machine learning models and automation for product teams and enterprises. Offers platforms and integrations aimed at improving model reliability, iteration speed, and production readiness.
Industry: Developer Tools
Website