AI Engineer

Ironclad•San Francisco, CA

8d•Hybrid

About The Position

Ironclad is accelerating its AI initiatives to transform how legal teams manage contracts. We're looking for a AI/ML Engineer to help shape the future of legal AI. In this role, you’ll work with cutting-edge tools such as HuggingFace, OpenAI APIs, and Rivet to develop state-of-the-art models and intelligent systems that extract structured insights from contracts and power product features used by leading legal teams. You’ll partner closely with product managers and designers to translate user needs into AI-powered product capabilities. You’ll also lead efforts in model evaluation, iteration, and deployment—ensuring our systems are robust, explainable, and continuously improving. As part of our end-to-end ownership model, you’ll contribute to the MLOps stack to ensure our models are scalable, reliable, and performant in production.

Requirements

BS degree in Computer Science, Engineering, or a related field
Experience working with large language models (e.g., GPT, Claude) or building RAG systems
Experience with embedding models and vector databases
Experience in training, evaluating, and serving machine learning models
Hands-on experience with Python, backend service development, and RESTful APIs
Experience in NLP techniques such as text classification, entity extraction, document understanding, or question answering
Proficiency with AI frameworks and APIs (e.g., PyTorch, TensorFlow, HuggingFace , OpenAI , LangChain)
Experience with cloud platforms (e.g., GCP, AWS, Azure)
Solid understanding of Git or other version control systems
Strong engineering fundamentals and attention to code quality, testing, and maintainability
Excellent communication and collaboration skills
Passion for building AI products that solve real-world customer problemsS

Nice To Haves

Familiarity with Vertex AI or other managed ML platforms
Experience working with PDF and DOCX parsing for document ingestion
Comfort with Docker and containerized development
Familiarity with CI/CD systems (e.g., GitHub Actions, CircleCI)
Monitoring and observability with tools like Datadog or equivalent
Experience with data processing and analysis using tools like SQL, Pandas, NumPy, or PySpark

Responsibilities

Build and Tune models and AI capabilities that ingest legal documents, extract key information, and deliver actionable insights to customers
Implementing backend services that power document understanding, classification, and natural language search
Explore and apply cutting-edge techniques in NLP, LLM integration, Prompt / Context Engineering, and semantic search to advance Ironclad’s AI/ML vision
Design and maintain robust systems for prompt management, model training / tuning, orchestration, evaluation & inference, with a focus on scalability, reliability, and observability
Develop and scale retrieval-augmented generation (RAG) systems and agent-based orchestration frameworks for complex legal workflows
Help build foundational AI/ML infrastructure, including tools for experimentation, model versioning, and online/offline performance monitoring
Partner closely with product managers and designers to shape user-facing AI experiences and deliver value through APIs and intelligent features

Benefits

100% health coverage for employees (medical, dental, and vision), and 75% coverage for dependents with buy-up plan options available
Market-leading leave policies, including gender-neutral parental leave and compassionate leave
Family forming support through Maven for you and your partner
Paid time off - take the time you need, when you need it
Monthly stipends for wellbeing, hybrid work, and (if applicable) cell phone use
Mental health support through Modern Health, including therapy, coaching, and digital tools
Pre-tax commuter benefits (US Employees)
401(k) plan with Fidelity with employer match (US Employees)
Regular team events to connect, recharge, and have fun
And most importantly: the opportunity to help build the company you want to work at