Lead Data Scientist – Generative AI

Requirements

Must have:

### We require formal education in machine learning, covering topics such as dimensionality reduction, clustering, embeddings, and sequence classification algorithms. You should have experience with deep learning frameworks like PyTorch, TensorFlow, and Hugging Face Transformers. Familiarity with Natural Language Processing techniques and libraries, including spaCy, word2vec, TensorFlow, Keras, PyTorch, Flair, and BERT, is essential. Furthermore, practical knowledge of large language models, prompt engineering, fine-tuning, and benchmarking with frameworks like LangChain and LlamaIndex is expected. A solid background in Python is necessary, along with an understanding of data modeling principles and complex data structures. Proficiency in both relational and NoSQL databases, as well as vector storage solutions such as Postgres and Elasticsearch/OpenSearch, is vital. Experience with Scala, Spark, Ray, or other distributed computing systems is highly preferred. Additionally, knowledge of API development, containerization, and machine learning deployment is advantageous, as is experience with ML Ops or AI Ops.

Responsibilities:

In this role, you will develop and implement applications based on large language models specifically designed for our internal legal team. You’ll fine-tune and deploy these models to improve their efficiency in processing legal texts. Your responsibilities will also include evaluating and maintaining our data assets and training/evaluation data sets, alongside designing and constructing pipelines for preprocessing, annotating, and managing legal document datasets. You’ll collaborate closely with legal experts to comprehend their requirements and ensure our models align with specific domain needs. Conducting experiments and assessing model performance will be key to fostering ongoing improvements. You will work together with other technical personnel and team members to finalize requirements, translating moderately complex product specifications into effective software designs. Additionally, you will implement development methodologies, coding best practices, and code reviews suited for production settings. ###

Company:

Applications will be accepted until December 8, 2025. We will only consider applicants for this position who are legally authorized to work in the United States without company sponsorship. Please be aware that this role is unable to offer visa transfer or sponsorship now or in the future. The annual salary for this position ranges from $120,000 to $160,000 based on experience and other qualifications. This role is also eligible for our discretionary annual incentive program, based on performance and terms specified in our applicable plans. We offer a variety of benefits for this position, contingent upon eligibility, including medical, dental, and vision insurance, paid holidays and time off, a 401(k) plan with contributions, long and short-term disability insurance, paid parental leave, and an employee stock purchase plan. Please note that the salary, compensation, and benefits information is accurate as of the date of this posting, and we reserve the right to modify it at any time in compliance with applicable laws.