Member of Technical Staff - GPU Engineer
5 days ago
Description
• We are seeking an experienced GPU Engineer with a strong background in Python and large-scale model training. • In this role you will design and implement improvements to our large scale training infrastructure and directly help make technical decisions to optimise our models' training performance and efficiency. • Reka's mission is to build useful multimodal artificial intelligence and use it to empower organisations and businesses.
Requirements
• Strong engineering skills with fluency in Python and PyTorch or other acceleration libraries. • Experience writing and debugging low-level GPU code (CUDA, Triton) and debugging hardware errors. • Experienced in scaling up GPU jobs via large-scale compute clusters using Slurm or Kubernetes. • Preferred: knowledge of advanced filesystems, particularly Ceph, and their integration and optimization in large systems. • Preferred: proficient in implementing robust monitoring systems for performance tracking and anomaly detection.
Benefits
• 4 weeks paid leave. • Visa support (such as H1B and OPT transfer for US Employees). • Healthcare benefits, including vision and dental.
Apply NowSimilar Jobs
6 days ago
Ventana Micro Systems
51 - 200
Develop and optimize compilers for cutting-edge RISC-V and chiplet technology.
6 days ago
Ventana Micro Systems
51 - 200
Develop frontend design workflows for semiconductor innovations at Ventana Micro Systems.
6 days ago
Canadian Solar Inc.
10,000+
Lead BESS implementation for Canadian Solar's energy storage solutions.