Senior Machine Learning Engineer

June 3

Apply Now

Receive Emails with Similar Jobs

Wizard AI

WebsiteLinkedInAll Job Openings

51 - 200

💰 $50M Series A on 2021-10

Description

• Build and maintain a training pipeline for fine-tuning open-source LLMs. Testing different LLM architectures at different sizes and quantizations is important. • Serve and Deploy the LLM on GPUs. It is important to know different configurations ahead of time to ensure we don’t OOM at runtime. • Formulate novel product issues to ML problems and propose and iterate on solutions quickly. We value the fast pace of development. • Triage customer issues to model/system shortcomings and communicate the complex architecture in a simple, comprehensible way to internal stakeholders. • Keep up with the fast-growing generative AI space and incorporate the latest developments into our system.

Requirements

• MS/Ph.D. in computer science, mathematics, or another quantitative discipline or exceptional work experience in ML engineering • 5+ years ML Experience • Experience developing machine learning-driven products at scale • Experience solving problems using Machine Learning with PyTorch • Exceptional understanding of designing ML systems in production at scale and applying a quantitative approach to identifying performance bottlenecks and cost/performance tradeoffs • Practical experience with large data systems, data models, and batch and streaming data pipelines • Ability to ramp up quickly on our tech stack, which includes GCP, Kubernetes, Airflow, Pandas, PyTorch, Python, and Node.js • Extensive experience with model deployment, both static within applications and dynamic, using Flask, FastAPI, or similar frameworks

Benefits

• Early-stage startup with massive growth potential and ability to grow as Wizard grows • Competitive compensation packages, including equity • Health: Comprehensive, high-quality medical coverage Dental & vision insurance OneMedical memberships for you and dependents Spring health platform for mental healthcare personalized to your needs XP Health eyewear benefits ($180, 3x per year) Rightway Health Guide • Wealth: 401(k) Plan Life & Disability insurance covered by Wizard • Work/Life: Flexible PTO and sick time to take care of yourself and your family 12 paid holidays 16 weeks parental leave for primary and secondary caregivers