
Data Scientist at AI Squared (Washington, DC)
AI Squared· Washington, DC·
Job description
Data Scientist
Washington, DC (Hybrid)
About the Role:
We are looking for a highly motivated Data Scientist with a strong background in applied machine learning and AI to join our growing team. In this role, you will be a key contributor to the development of core AI/ML solutions that power our platform. You will collaborate closely with product and engineering teams, applying state-of-the-art techniques to solve complex challenges, advance our use of large language models (LLMs), and ensure scalable, production-ready solutions.
Key Responsibilities:
- Leverage 5+ years of experience in data science to design, implement, and optimize machine learning models and pipelines.
- Develop, fine-tune, and evaluate large language models (LLMs) for a variety of applications, ensuring accuracy, performance, and robustness.
- Collaborate with engineering and product teams to integrate AI/ML solutions into our platform in a scalable and maintainable way.
- Conduct applied research, staying current on advances in LLMs, generative AI, and data science methodologies, and translate them into practical solutions.
- Build end-to-end workflows, from data exploration and feature engineering to training, validation, deployment, and monitoring in production.
- Apply modern containerization and orchestration techniques (e.g., Docker, Kubernetes) to support reproducible experimentation and deployment.
- Work with cloud platforms (e.g., Databricks, AWS, GCP, Azure) to manage data pipelines, large-scale training jobs, and distributed systems.
- Collaborate across teams to ensure our AI capabilities align with platform goals and business needs.
Qualifications:
- 5+ years of experience as a Data Scientist or Machine Learning Engineer, with proven success in deploying models to production.
- Hands-on experience with large language models (LLMs); fine-tuning experience strongly preferred.
- Strong background in Python and ML frameworks such as PyTorch or TensorFlow.
- Proficiency in containerization and orchestration technologies (Docker, Kubernetes).
- Experience with cloud platforms and ML ecosystems (Databricks, AWS, GCP, Azure).
- Familiarity with MLOps best practices, including model deployment, monitoring, and CI/CD for ML.
- Strong analytical and problem-solving skills, with the ability to translate research into production-ready solutions.
- Excellent communication and collaboration skills, with the ability to work effectively across product, engineering, and leadership teams.
- A proactive, self-starter mindset with a passion for applied research and innovation.
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.