Senior Machine Learning Engineer - BLaw/BTax/BGov

About the Role

Key Responsibilities

• Drive, design & develop ML projects as the principal point-of-contact
• Collaborate with domain experts and product managers to understand business needs and map business problems to ML problems
• Research groundbreaking ML/NLP techniques and apply them to business problems
• Develop tailored ML/AI prediction models for the domain
• Use metrics to make data-driven decisions
• Write and maintain production-quality code
• Collaborate with AI platform engineers on model maintenance and rollout
• Manage stakeholders' expectations throughout the project on model development and release

Requirements

• 4+ years of experience in Python
• Proven delivery of production NLP systems
• Intuition to formulate an ML problem from a business problem
• Master's or PhD in Computer Science, Engineering, Mathematics, or similar field of study or equivalent work experience

Nice to Have

• Knowledge of advanced concepts such as weakly supervised learning, reinforcement learning, semantic search, knowledge-graph construction
• Authored research publications, participation in ML competitions, working demos/repos
• Experience with distributed computational frameworks (Spark, Hadoop, Kubernetes, Databricks)
• Curiosity to learn more about the legal, tax or government domain
• Familiarity with traditional statistical methods, modern deep learning frameworks, prompting LLMs, RAG, vector search
• Experience in all phases of machine learning application lifecycles from problem mapping and scoping to data gathering and preparation to optimizing model performance

Benefits & Perks

• Salary range of $165,000 - $260,000 USD annually plus benefits and bonus
• Comprehensive and generous benefits plan including merit increases, incentive compensation, paid holidays, paid time off, medical, dental, vision, short and long term disability benefits, 401(k) with match, life insurance, and wellness programs