6 months ago
Remote, Worldwide +3 moreMid Level / Senior
H1B Sponsor
Responsibilities
- Design and implement improvements to training infrastructure.
- Optimize performance of models through technical decision-making.
- Work on post-training processes including reinforcement learning and fine-tuning.
- Enhance efficiency and scalability of model serving infrastructure.
Requirements
- 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 performance of GPU-accelerated workloads.
Benefits
- Five weeks of paid leave to recharge.
- Comprehensive healthcare benefits including vision and dental.
- Inclusive and open work culture that values diverse perspectives.
- Visa assistance for US employees including H1B and OPT transfers.