Senior Machine Learning Engineer

Summary

Join Niche as a Senior ML Engineer and design, develop, and deploy scalable machine learning solutions for AI Assistants, Search, and Recommendations. You will leverage your expertise in MLOps, natural language processing, and deep learning to enhance user and partner experiences. Key responsibilities include AI Assistant/Chatbot development, resource management, search and recommendation optimization, MLOps oversight, rapid prototyping, system integration, and technical leadership. The ideal candidate possesses a Master's degree in a relevant field, 5+ years of ML engineering experience (including 3+ years with AI Assistants), and proficiency in Python and various ML libraries. Competitive compensation, including a base salary range of $127,670-$159,589 plus bonuses and stock options, is offered along with a comprehensive benefits package.

Requirements

• Master's degree in Computer Science, Machine Learning or a relevant engineering discipline
• 5+ years of experience in machine learning engineering, with 3+ years of demonstrable experience in working with AI Assistants
• Proficiency in Python, TensorFlow, PyTorch, Keras, scikit-learn and relevant libraries. Hands-on experience in MLOps/LLMOps, observability and CICD pipelines
• Strong understanding of NLP, deep learning architectures, recommendation systems and evaluation metrics
• Ability to tackle complex technical problems with strong analytical skills
• Excellent verbal and written communication abilities

Responsibilities

• Engineer ML pipelines to optimize conversational AI applications that leverage deep learning frameworks including transformer architectures
• Optimize GPU/TPU and memory usage to optimize compute costs of ML pipelines while ensuring adherence to established SLAs
• Deep understanding of concepts like semantic search, query understanding, relevance and ranking, candidate generation
• Oversee data preprocessing, feature engineering, ML observability and model registry to ensure quality of outputs from AIML applications
• Ability to quickly iterate through ML System Designs and prototype ML solutions for continuous stakeholder feedback
• Collaborate with software engineers to integrate ML models into production systems, ensuring seamless user experiences
• Mentor junior engineers, partner with software engineers to provide technical guidance and foster a culture of innovation

Preferred Qualifications

• Experience with cloud computing platforms (e.g., AWS, Google Cloud, Azure)
• Familiarity with Llamaindex, LangChain, Ragas and agentic design patterns
• Knowledge of search engine technologies e.g., Elasticsearch, OpenSearch and vector databases
• Experience with RAG architectures and ML pipelines for reinforcement learning
• Familiarity with agile development methodologies

Benefits

• Our national target base salary range is $127,670-$159,589, plus participation in our Annual Bonus and Stock Option Program. Base compensation will be commensurate with experience and skills
• Best-in-class 100% paid employee health plan, including vision and dental and supplemental coverage
• Flexible Paid Time Off Policy
• Stipend that allows you to build your work from home office in a style and function that suits your personal preferences
• Parental leave for all employees (12 weeks fully paid) in addition to short term disability for birthing parents
• Meaningful 401(k) with employer match
• We are a fully flexible workforce empowering our employees to choose to work remotely, in our Pittsburgh office or whatever combination suits you