lever
Senior Machine Learning Engineer
Tahoebio-ai
South San Francisco, USA
Full-time
Senior
Onsite
Discovered 1 months ago
PyTorchTensorFlowKerasJAXHF AccelerateDeepSpeed
Key Skills for This Role
PyTorchTensorFlowKerasJAXHF AccelerateDeepSpeed
Full Job Posting
- 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
- 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.
- 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.
- 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.
About Tahoebio-ai
Verified company details for this employer are not available yet.
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer