Senior AI/ML Engineer

Cooley AI • Washington, DC

1d • $145,000 - $210,000

About The Position

Cooley is seeking a Senior AI/ML Engineer to join its Practice Engineering team. Cooley Innovation aims to advance the digital practice of law through customer service excellence. This hands-on, execution-focused role builds, deploys, and operates enterprise-scale AI and machine learning solutions across Azure, AWS, and other cloud and hybrid platforms. The engineer applies software engineering practices and AI/ML expertise to deliver secure, scalable, high-performing solutions that support the firm’s business and technology objectives.

Requirements

• At least 5 years of directly applicable experience, such as building and deploying AI and ML solutions.
• Hands-on experience developing AI and ML solutions on Azure and/or AWS.
• Strong proficiency in Python for building and operating LLM-powered applications and agentic workflows.
• Experience integrating LLM applications with enterprise data sources, APIs, and retrieval systems (RAG).
• Hands-on experience building agentic and LLM-powered applications using frameworks such as LangChain, LangSmith, Pydantic AI, or similar orchestration and observability tools.
• Knowledge of cloud security, governance, and data privacy principles as they relate to AI and ML systems.
• After orientation at Cooley LLP, proficiency with Microsoft Office, iManage, and other firm applications.
• Ability to work extended and/or weekend hours as required.
• Ability to travel as required.
• Excellent oral and written communication skills, including technical and user documentation writing.
• Strong analytical, problem-solving, critical-thinking, and implementation skills.
• Professional demeanor and high level of professionalism.
• Excellent attention to detail, follow-up, organizational, and communication skills.
• Ability to work independently and under pressure, and to solve problems independently and simultaneously.
• Ability to organize, prioritize, and coordinate multiple activities, often under tight timelines.
• Ability to drive projects to completion and achieve goals.
• Strong judgment and a collaborative, team-oriented spirit.
• Ability to handle and maintain confidentiality regarding firm information, projects, and client data.
• Effective presentation skills.

Nice To Haves

• Bachelor’s degree in Computer Science, Engineering, Information Systems, or a related discipline.
• Entrepreneurial nature.

Responsibilities

• Design, build, and maintain production-grade AI and ML solutions across Azure, AWS, and other cloud platforms.
• Implement end-to-end AI application pipelines, including data ingestion, prompt engineering, evaluation, deployment, and monitoring.
• Collaborate with engineering, data, and business teams to translate requirements into practical AI and ML implementations.
• Write high-quality, scalable, and maintainable code for AI and ML workloads.
• Optimize performance, reliability, and cost efficiency of cloud-based AI and ML services.
• Integrate AI and ML solutions with enterprise data platforms, including data lakes, data warehouses, APIs, and streaming systems.
• Support AI solution lifecycle management, including versioning, prompt iteration, evaluation, and production support.
• Apply security, compliance, and data privacy best practices in AI and ML implementations.
• Contribute to technical design discussions and provide guidance on AI and ML implementation patterns.
• Stay current with AI, ML, and cloud technology advancements and apply them pragmatically.
• Perform other duties as assigned or required.

Benefits

• Competitive compensation and an excellent benefits package.
• Medical benefits.
• Health savings account with applicable medical plan.
• Dental insurance.
• Vision insurance.
• Health and/or dependent care flexible spending accounts.
• Pre-tax commuter benefits.
• Life insurance and accidental death and dismemberment coverage.
• Long-term care coverage.
• Backup care for children and/or adults and other parental support benefits.
• Firm-paid life insurance, accidental death and dismemberment coverage, long-term disability, and short-term medical benefits for benefited employees.
• 21 days of paid time off each year.
• 10 paid holidays each year.
• Generous parental leave.
• Fertility benefits.
• Detailed benefits orientation for new employees.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed