Member of Technical Staff (ML)

About the role

As a Member of Technical Staff focusing on Machine Learning, you will join a globally distributed team building cutting‑edge multimodal AI models. You will work in a fast‑paced, remote‑first environment, collaborating with top engineers and researchers to turn research breakthroughs into practical products.

Key responsibilities

• Contribute to the full development cycle of large deep learning models, from data preparation to deployment.
• Design model architectures, implement solutions, train and evaluate models to improve product performance.
• Collaborate closely with engineers and researchers to translate cutting‑edge research into practical applications.
• Wear multiple hats, helping to build systems from the ground up in a dynamic startup environment.

Required profile

• Experience training and evaluating large deep learning models.
• Proficiency with standard deep learning frameworks such as PyTorch or JAX.
• Proven track record of applying machine‑learning algorithms at scale.
• Ability to thrive in a fast‑paced environment with some uncertainty.
• Kind, collaborative nature and work style.

Required skills

• PyTorch
• JAX

What we offer

• 5 weeks of paid leave to recharge.
• Comprehensive healthcare benefits, including vision and dental coverage.
• Visa assistance, including H1B and OPT transfers for US employees.