Senior Machine Learning Engineer – NLP & Search
| Company | Ontra |
|---|---|
| Location | San Francisco, CA, USA, Santa Barbara, CA, USA |
| Salary | $176000 – $264000 |
| Type | Full-Time |
| Degrees | Bachelor’s, Master’s |
| Experience Level | Senior |
Requirements
- 4+ years in applied machine learning or NLP roles
- Proficient understanding of transformer-based architectures, embeddings, and document-level NLP
- Demonstrated experience in deploying ML models in production environments, including the use of orchestration and monitoring tools
- Familiarity with semantic search, vector stores (e.g., Pinecone, Weaviate), or hybrid RAG systems
Responsibilities
- Design, build, and scale advanced machine learning systems to automate the extraction and retrieval of legal obligations from complex private equity documents
- Develop and fine-tune LLM solutions for legal workflows, enhancing the capabilities of Ontra’s AI systems with custom NER pipelines and semantic indexing
- Work closely with product, legal, and engineering teams to define project requirements and seamlessly integrate AI solutions into existing workflows
- Review and refine datasets and machine learning pipelines to ensure they deliver high performance, scalability, and are tailored to evolving needs
- Deploy and continuously monitor machine learning models in production, conducting experiments and implementing optimizations to improve model performance
- Drive measurable improvements in legal review processes and enhance end-user productivity through advanced AI applications and tools
Preferred Qualifications
- Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field preferred