Reka
Member of Technical Staff (GPU Performance Engineer)
US, UK, Singapore, Remote. Posted 10 months ago
The role
You will design and implement improvements to training infrastructure and contribute to technical decisions to optimize model performance. You will work on post-training processes including reinforcement learning and fine-tuning and improve model serving infrastructure efficiency and scalability.
What you will do
- Design and implement improvements to training infrastructure
- Contribute to technical decisions to optimize model performance
- Work on post-training processes including reinforcement learning and fine-tuning
- Improve the efficiency and scalability of model serving infrastructure
Required qualifications
- Fluency in Python and PyTorch (or other frameworks)
- 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
Benefits
- Five weeks of paid leave
- Comprehensive healthcare benefits including vision and dental
- Additional perks that support well-being
Before you apply
- Visa assistance including H1B and OPT transfers is provided for US employees
About the company
Reka is a globally distributed foundation model startup headquartered in the San Francisco Bay Area, California. The mission is to build useful multimodal artificial intelligence to empower organizations and businesses.