AI Squared

Machine Learning Engineer at AI Squared (Washington, DC)

AI Squared· Washington, DC·

Job description

Machine Learning Engineer

Washington, DC (Hybrid)

About the Role:

We are seeking a highly skilled Machine Learning Engineer to join our core AI team. In this role, you will focus on deploying, maintaining, and monitoring the AI/ML systems that power our platform. You will work closely with data scientists, data engineers, and product teams to ensure scalable, reliable, and production-grade AI solutions. You’ll play a critical role in operationalizing large language models (LLMs) and other ML systems, ensuring they run efficiently, securely, and with robust monitoring in place.

Key Responsibilities:

  • Design, implement, and maintain ML deployment pipelines for scalable production systems.
  • Operationalize large language models (LLMs) and other AI/ML models, ensuring high availability and reliability.
  • Build robust model monitoring, logging, and alerting systems to track performance and detect drift.
  • Partner with data scientists to transition models from research/prototype into production-ready deployments.
  • Develop CI/CD pipelines for ML workflows, integrating testing, validation, and automated deployment.
  • Optimize runtime performance of ML models across cloud platforms (AWS, GCP, Azure) and distributed systems.
  • Apply containerization and orchestration (Docker, Kubernetes) to enable reproducible, scalable systems.
  • Collaborate with cross-functional teams to ensure ML systems align with platform goals and business requirements.

Qualifications:

  • 5+ years of experience as a Machine Learning Engineer, MLOps Engineer, or similar role.
  • Proven experience deploying and maintaining machine learning models in production at scale.
  • Hands-on experience with ML lifecycle tooling (MLflow, Kubeflow, SageMaker, Vertex AI, or similar).
  • Strong proficiency in Python; familiarity with ML frameworks such as PyTorch or TensorFlow.
  • Deep knowledge of containerization (Docker) and orchestration (Kubernetes) for production ML systems.
  • Expertise with cloud platforms (AWS, GCP, Azure) for ML deployment and scaling.
  • Strong understanding of MLOps best practices, monitoring, and automation.
  • Excellent problem-solving skills, with an emphasis on building reliable, scalable systems.
  • Strong communication and collaboration skills across technical and non-technical teams.

Why work at AI Squared

  • Culture: Mission-driven, innovation-first, with a focus on "impact over hype." Values include relentless curiosity and security & trust.
  • Hybrid Work Policy: Flexible model empowering employees to work from home or in offices, with in-person collaboration for key moments of innovation.
  • Offices: Two vibrant hubs — Mountain View, CA (West Coast HQ for engineering, product, and AI research) and Bangalore, India (cutting-edge AI infrastructure development). Also has a Washington, DC office for federal work.
  • Benefits: Comprehensive health coverage (medical, dental, vision), equity & competitive compensation, 401(k) matching, generous unlimited PTO, paid parental leave, wellness reimbursements, and employee assistance programs.
  • Engineering Culture: Focus on building the infrastructure for agentic enterprise AI with a small, fast-growing team (52 employees). Open roles include Backend Engineer, Machine Learning Engineer, Data Scientist, and QA Engineer.

Application questions