Staff Machine Learning Engineer, Gen AI

Weave is looking for engineers hungry for fun challenges who can join our self-empowered teams and contribute in both technical and non-technical ways. You will be joining a team of talented developers that share a common interest in distributed backend systems, data, scalability, and continued development.

Requirements

• 8+ years of experience in Machine Learning or AI, preferably with a focus on natural language.
• Experience moving and storing TBs of data or 100M’s to 10B’s of records.
• Experience building and deploying ML driven B2B multi-tenant applications in production environments at scale for external products and customers.
• Experience with common ML technologies such as Python, Jupyter, Workflow Engines (Dagster, MLFlow, KubeFlow, etc), DVC, Triton Server, LLMs, Postgres, and others.
• Experience with modern ML tools, techniques, and evaluation such as LLMs, RAG, Prompt Engineering, Fine Tuning, LLM evaluations, multi-modal models, and others.
• Experience with data labelling or annotation for audio or text use cases.
• Understanding of distributed systems and building scalable, redundant, and observable services.
• Expertise in designing and architecting systems for distributed data sets and services.
• Experience building solutions to run on one or more of the public clouds (e.g., AWS, GCP, etc.).
• Experience providing stable well designed libraries and SDKs for internal use.
• Self driven and a thirst for learning in a quickly changing industry.
• Demonstrated track record of delivering complex projects on time and have experience working in enterprise-grade production environments.
• Demonstrated experience working with varied stakeholders across multiple teams.
• Demonstrated leadership experience leading large projects or teams.
• Strategic thinker with a strong technical aptitude and a passion for execution.

Benefits

• Competitive salary
• Stock options
• Health insurance
• Retirement plan
• Paid time off
• Generous parental leave