Salary
💰 $210,000 - $235,000 per year
Tech Stack
AWSAzureCloudDockerGoogle Cloud PlatformNumpyPandasPySparkPythonPyTorchSQLTensorflow
About the role
- 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
- 2 years of previous experience exploring and applying cutting-edge techniques in NLP, LLM integration, Prompt / Context Engineering, and semantic search to advance Ironclad’s AI/ML vision
- 2 years of previous experience designing and maintaining robust systems for context management, model training / tuning, orchestration, evaluation & inference, with a focus on scalability, reliability, and observability
- 2 years of previous experience developing and scaling 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
Requirements
- BS or advanced 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 AI/ML models
- 7+ years of 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 problems