Senior AI Engineer
iManage•Chicago, IL
•$180,000 - $220,000•Hybrid
About The Position
We are seeking a passionate Senior AI Engineer to join our Applied AI team. In this role, you will be responsible for building and deploying AI systems at scale in production. You will collaborate with engineers, data scientists, and product stakeholders to develop AI capabilities for iManage’s enterprise work platform. This includes working on generative AI document assistants, document classification, extraction, and LLM-driven features that enhance how knowledge workers manage their information. The ideal candidate will have hands-on experience in both model development and the infrastructure required for reliable deployment, thriving at the intersection of machine learning, software engineering, and cloud infrastructure.
Requirements
- A Bachelor’s, Master’s, or Ph.D. in Computer Science, Machine Learning, Data Science, Artificial Intelligence, Statistics, or a related field
- 3+ years of experience in ML/AI engineering or software engineering
- Hands-on experience building and shipping LLM systems into production
- Deep proficiency in Python and modern AI frameworks, including PyTorch and Hugging Face
- Solid understanding of ML fundamentals including hands-on experience with both traditional ML and modern generative AI and transformer architectures
- Demonstrated experience fine-tuning language models and deploying them to production
- Experience with GPU optimization for training and inference workloads
- Experience with Kubernetes deployment on cloud infrastructure (Azure, AWS, or GCP) to optimize systems for scalability, latency, performance, and cost efficiency
- The ability to work collaboratively across teams, communicate with precision, and take ownership from prototype to production
Nice To Haves
- Experience with distributed training frameworks (e.g., PyTorch Distributed, Ray) and inference optimization tools (e.g., vLLM, SGLang)
- Experience with agentic engineering, including agent harness and agent memory, and orchestration frameworks such as LangChain, LlamaIndex, or similar tools
- Experience with knowledge graphs and multimodal LLMs
- Familiarity with AI/ML observability tools and model lifecycle management best practices
Responsibilities
- Owning the end-to-end ML lifecycle for AI systems from model development and evaluation through to scalable production serving, generative AI document intelligence, and agentic system use cases
- Deploying and optimizing ML/AI systems on GPUs and Kubernetes-based cloud infrastructure, including AKS or equivalent platforms, while balancing trade-offs across scalability, latency, performance, operational complexity, and cost efficiency
- Designing and implementing production-ready LLM applications and APIs with monitoring, observability, and integration testing built in
- Applying modern engineering practices for production AI systems, including containerized services, CI/CD pipelines, model and version tracking, and release governance
- Collaborating with product and business stakeholders to translate requirements into viable technical solutions
- Conducting code reviews and providing constructive feedback to team members
- Contributing to best practices and standards for AI engineering across the team
Benefits
- Flexible working policy
- Flexible work hours
- Supportive, experienced team with an inclusive, encouraging, and vibrant culture
- Modern open plan workspace with a gaming area, free snacks, drinks and regular social events
- Focus on impactful work, solving complex, real challenges utilizing the latest technologies and protocols
- Internal development framework for career path ownership
- Access to unlimited courses in LinkedIn Learning
- Market competitive salary
- Annual performance-based bonus
- Comprehensive Health/Vision/Dental/Life Insurance
- 401k Retirement Savings Plan with a company match up to 4%
- HealthJoy, a healthcare concierge service
- Enhanced leave for expecting parents (20 weeks 100% paid for primary leave, and 10 weeks 100% paid for secondary leave)
- Flexible time off policy
- Company wellness days
- Free access to the Healthy Minds app
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
Upload and Match Resume
What This Job Offers
Job Type
Full-time
Career Level
Senior
