Open roles that use PyTorch for machine learning.
3
open jobs
Salary: £65,000 - 85,000 per year Requirements: Knowledge of C++, Python, PyTorch, OpenCV, CUDA, TensorRT, Git, WSL, Docker, and Jira/Confluence Machine Learning and Deep Learning Computer Vision methodologies and…
Own model training and post-training pipelines end to end: SFT, RLHF, PPO, DPO, and reward model training in PyTorch Build and maintain the infrastructure around RL training: rollout collection, data curation, reward…
The Product: AWS Machine Learning accelerators are at the forefront of AWS innovation. Trainium delivers best-in-class ML training performance with the most teraflops (TFLOPS) of compute power for ML in the cloud. This…
We couldn't check your fit for this role — add a CV to your profile to see it next time.