About the role
As a Machine Learning Engineer at WPMaven, you'll develop, train, and optimize the models the operators depend on: understanding a business from a description, writing in its voice, and generating the code and configuration behind a working store. You'll work at the intersection of ML research and production engineering.
Responsibilities
- Design and implement models for the work the operators do: content production, understanding a business, and generating the code behind a working store.
- Fine-tune large language models for the tasks the operator team runs every week.
- Develop data pipelines for collecting, cleaning, and processing training data.
- Implement evaluation frameworks to measure model quality and identify areas for improvement.
- Collaborate with product teams to understand user needs and translate them into ML solutions.
- Optimize models for production deployment, balancing accuracy with latency and cost.
- Stay current with ML research and identify techniques applicable to our domain.
- Document models, experiments, and findings for team knowledge sharing.
Requirements
Experience
- 3+ years of experience building and deploying ML models.
- Experience with NLP and/or code generation tasks.
- Track record of taking ML projects from research to production.
Technical skills
- Strong Python programming skills and experience with ML frameworks (PyTorch preferred).
- Experience with transformer architectures and fine-tuning LLMs.
- Familiarity with ML experiment tracking and model versioning.
- Understanding of ML evaluation metrics and A/B testing.
- Experience with distributed training and GPU optimization.
Nice to have
- Experience with code generation or code understanding models.
- Publications in relevant ML conferences (NeurIPS, ICML, ACL, etc.).
- Experience with retrieval-augmented generation (RAG) systems.