Note: The job is a remote job and is open to candidates in USA. The ReWork Group partners with high-growth startups and forward-thinking companies to build the future. They are seeking a Principal Machine Learning Engineer who will design and build production-grade AI agents and own the full lifecycle of machine learning models for enterprise-scale applications.
Responsibilities
- Design and build production-grade AI agents using Python, LLMs, and agent orchestration frameworks that internal teams actually depend on
- Own the full lifecycle: from workflow analysis and architecture through implementation, deployment, monitoring, and iteration
- Design, build, and deploy ML models for demand forecasting, time-series prediction, consumer sentiment analysis, and anomaly detection — at enterprise scale
- Develop and iterate on our agentic AI architecture — systems that reason across heterogeneous data sources and take autonomous action
- Own robust ML pipelines end to end — data preprocessing, feature engineering, model training, evaluation, and production deployment
- Architect and sharpen production graph RAG system — one of their core technical differentiators
- Build RAG systems and LLM integrations that power natural-language interfaces and autonomous workflows
- Partner with backend engineers to make models genuinely production-grade — tuned for latency, reliability, and scale
- Own model performance in the wild — monitoring, retraining, and continuous improvement once it's live
- Stay at the frontier of AI research and pull the innovations that actually matter into the platform
Skills
- 8+ years of experience in applied machine learning and AI, with models deployed and running in production environments
- M.S. or Ph.D. in Computer Science, Machine Learning, Statistics, or related field (or equivalent practical experience — what you've built matters more than the degree)
- Deep proficiency in Python with experience in ML frameworks (PyTorch, TensorFlow, scikit-learn)
- Experience with NLP, LLMs, and RAG architectures
Company Overview