AI Engineer →

Nextlex; San Francisco (CA); Hybrid; USD 150,000 - 210,000; Full time

Job summary

Nextlex is seeking a seasoned ML Engineer to design, build, and deploy core AI models that power its intelligence layer. You will transform fragmented legal data into actionable insights, teaching the platform to understand and streamline complex workflows.

Responsibilities include developing NLP-focused models, managing data pipelines, and collaborating with researchers to scale prototypes into production. A strong grounding in MLOps, containers, and cloud platforms is essential.

Qualifications

• 3+ years of experience building and deploying ML/AI systems.
• Strong Python skills and experience with TensorFlow or PyTorch.
• Solid understanding of MLOps, containerization (Docker), and cloud platforms (AWS/GCP/Azure).
• Proven ability to handle large, unstructured datasets.

Responsibilities

• Develop, optimize, and maintain production-ready AI/ML models, especially NLP and information-extraction systems.
• Build and manage data pipelines for training, testing, and deploying large-scale models.
• Collaborate with AI Researchers to scale prototypes into production models.
• Ensure model interpretability, reliability, and security within a sensitive legal tech environment.

Skills

Python

NLP

ML deployment

Tools

TensorFlow

PyTorch

Docker

AWS

GCP

Azure

Job description

Design, build, and deploy the core machine learning models that power Nextlex's intelligence layer. This role transforms fragmented legal data into actionable insights, teaching the platform to understand, anticipate, and streamline complex legal workflows.

Key Responsibilities

• Develop, optimize, and maintain production-ready AI/ML models, especially NLP and information-extraction systems.
• Build and manage data pipelines for training, testing, and deploying large-scale models.
• Collaborate with AI Researchers to scale prototypes into production models.
• Ensure model interpretability, reliability, and security within a sensitive legal tech environment.

Qualifications

• 3+ years of experience building and deploying ML/AI systems.
• Strong Python skills and experience with TensorFlow or PyTorch.
• Solid understanding of MLOps, containerization (Docker), and cloud platforms (AWS/GCP/Azure).
• Proven ability to handle large, unstructured datasets.

Preferred Experience

• NLP specialization (summarization, topic modeling, relationship extraction).
• Knowledge of LLM architectures and fine-tuning.
• Familiarity with legal data types and compliance constraints.