Find your next role
Discover roles across ambitious teams
6,149 jobs
- More jobs
Reka
Member Of Technical Staff
Remote · Singapore, UK, USA · Full time · Mid level
About the role
Design and implement improvements to training infrastructure and optimize model performance. Work on post-training processes and improve model serving efficiency and scalability.
What you’ll do
• Design and implement training infrastructure improvements. • Contribute to technical decisions optimizing model performance. • Work on post-training processes including reinforcement learning. • Improve efficiency and scalability of model serving infrastructure.
What you’ll bring
• Fluency in Python and PyTorch or similar frameworks. • Experience training large deep learning models. • Experience writing and debugging GPU code (CUDA, C++). • Experience scaling GPU jobs with clusters like Slurm or Kubernetes. • Ability to profile and optimize GPU-accelerated workloads.
Life at Reka
Collaborate with elite engineers from top AI organizations. Enjoy five weeks paid leave, comprehensive healthcare, and visa support. Inclusive and open culture fostering creativity.
About Reka
Reka is a globally distributed foundation model startup building multimodal AI to empower organizations. Headquartered in San Francisco Bay Area with a remote-first team.