Member of Technical Staff, MLE
Build and deploy applied machine learning solutions for enterprise customers using large language models. The role requires experience leading ML teams, setting technical direction, working cross-functionally, and applying Python and modern ML frameworks to large-scale training and inference.
About the job
Responsibilities
- Plan and execute large-group projects from ideation through production.
- Align engineering, product, and other disciplines around shared goals.
- Mentor a distributed team of engineers in subject matter expertise.
- Identify opportunities and gaps in existing models and strategize areas of focus.
- Work closely with product teams to develop solutions.
- Collaborate with partner organizations.
- Assist legal teams with preparing patents for developed intellectual property.
Requirements
- Experience leading machine learning teams in applied research and production.
- Experience setting and executing technical direction to build new products or refine existing ones.
- Experience working cross-discipline in matrixed organizations.
- Proficiency in Python and related machine learning frameworks such as TensorFlow, TF-Serving, JAX, and XLA/MLIR.
- Experience with large-scale distributed training and inference.
- Strong mentorship, communication, and problem-solving skills.
- Demonstrated passion for applied machine learning models and products.
Nice-to-haves
- Experience working for B2B companies in the machine learning space.
Compensation and Benefits
- Weekly lunch stipend of $75/£75 or equivalent in local currency.
- Full health and dental benefits, including a separate mental-health budget.
- RRSP matching, 401(k), and pension scheme.
- 100% parental-leave top-up for up to six months for either parent.
- Annual enrichment benefits covering arts and culture, fitness and wellness, quality time, and workspace improvements.
- Education and learning stipend for conferences, courses, and coaching.
- Six weeks of paid vacation (30 working days).
- Budget for travel to other offices for remote employees and an annual company offsite.
- Daily lunch program, snacks, and community and social events for office employees.
- Co-working benefit for employees not near an office.
- $500 home-office stipend.
Skills
Python, TensorFlow, Tf-Serving, JAX, Xla, Mlir, Distributed Training, Distributed Inference, LLMs, Applied Machine Learning, Machine Learning Models, Production Ml
Similar jobs
ML Engineering jobs
Rollstack
United StatesBuild production AI capabilities for automated slide and document generation, working across LLM applications, data analysis, and content generation. The role requires 3+ years in machine learning and NLP, advanced Python, and experience with LLM frameworks and production systems.
OpenAI
London, United Kingdom
Build and deploy AI-powered products for digital-native customers, taking systems from experimentation through production and scale. The role requires strong Python skills, hands-on production engineering, systematic AI evaluation, and the ability to navigate reliability, security, governance, and customer impact.
Elliptic
London, United Kingdom
Build full-stack AI agent fleets, APIs, workflows, and internal services that automate complex business processes. The role requires at least five years of engineering experience, hands-on LLM framework experience, production AWS expertise, Kubernetes, and strong API and database skills.
Protege
Remote
Build the technical foundation for a new business vertical, creating reusable infrastructure and leading early customer engagements from scoping through delivery. The role requires 3+ years of engineering experience, strong Python and SQL skills, backend/data expertise, and comfort operating in ambiguity.
Build
New York, NYBuild and operate Dougie, an agentic AI system that executes workflows, evaluates its own performance, retains institutional context, and improves in production. The role requires experience deploying unattended agentic systems and engineering reliable memory, retrieval, orchestration, and feedback loops.