Prior experience in leading ML teams in applied research and production
Prior experience in setting as well as executing technical direction to build new or to refine existing products
Prior experience in working cross-discipline in matrixed organizations
Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR
Experience using large-scale distributed training and inference
Strong mentorship, communication, and problem-solving skills
A demonstrated passion for applied ML models and products
Bonus: experience working for B2B companies in the ML space
What the job involves
As a Member of Technical Staff on our Applied ML team, you will work directly with customers to quickly understand their greatest problems and design and implement solutions using Large Language Models