Wizard is on a mission to build the most powerful AI-driven conversational commerce platform. Our B2B software empowers brands to transact with consumers via text, launching an entirely new sales channel to capture increased transactions, conversions, engagement and loyalty. C-commerce is exploding to a $1 Trillion market in the next decade and Wizard is positioned to become the dominant technology in the space.
Wizard is backed by Marc Lore and a $50M Series A from top tech VCs. We’re quickly scaling our team and technology and seeking passionate and growth-minded tech talent to join the ride.
The Role
Wizard is hiring ML Engineers, preferably with a background in NLP, to work on all aspects of dialog systems, from fine-tuning LLMs to serving models and building data pipelines and entity resolutions. If you don’t have an NLP background but have deep experience with Python and some exposure to ML and you are excited about the space and willing to learn, we would still encourage you to apply.
Key Responsibilities:
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.
You:
MS/Ph.D. in computer science, mathematics, or another quantitative discipline or exceptional work experience in ML engineering.
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.
The expected salary for this role is $234,000-$260,000 depending on skills and experiences.
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
11 paid holidays
16 weeks parental leave for primary and secondary caregivers
In Office Perks (for NYC based employees)
Hybrid work environment
Monthly team outings
Snacks and drinks to keep you fueled throughout the day
Wizard is committed to building the best team and strongly believes that fair hiring practices will help us achieve our goals. We advocate for diversity, equity and inclusion because it’s not only the right thing to do, but also creates a positive work culture and stronger outcomes. Wizard does not discriminate or make decisions based on race, color, religion, gender identity or expression, sexual orientation, national origin, age, military service eligibility, veteran status, marital status, disability or any other protected class(es). If you have a disability please let our recruiting team know how we can make the interview process work for you.