Senior Machine Learning Engineer
Vevo Therapeutics
Software Engineering
South San Francisco, CA, USA
Qualifications - Required
- Solid Engineering and Computer Science fundamentals, ideally with a degree in CS, Math, or equivalent experience.
- Exceptional engineering skills to iterate quickly on data processing and training pipelines
- Experience in building, testing, training, and deploying modern neural network architectures such as Transformers.
- Experience with frameworks like PyTorch, Tensorflow, Keras, JAX and the ability to write hardware optimized code for distributed training
Qualifications - Desirable
- Experience with distributed deep learning using frameworks such as HF Accelerate, Deepspeed, Composer, TorchTitan, Megatron etc.
- Experience with training large neural networks on multiple compute nodes.
Key Responsibilities
- Stay at the forefront of ML and computational biology research and rapidly adopt state-of-the-art techniques to train AI virtual cell models on our massive single cell datasets.
- Work together with ML Scientists to develop the next generation of compute optimized models like Tahoe-x1.
Benefits
- Unlimited Paid Time Off (PTO).
- Monthly Lunch budget
- One-time Office set up budget
- US Employees: HMO Kaiser Platinum and PPO Anthem Gold medical as well as vision and dental plans for both the employee and dependents.