AI/ML Architect

Summary

Join BambooHR as an AI/ML Architect and work on an award-winning product with a talented team. You will architect, design, and implement scalable AI/ML systems, design architectural advancements, define incremental architectural changes, and collaborate with stakeholders to determine priorities. Responsibilities include building, fixing, and improving code, training team members, helping build AI-integrated code, defining model governance workflows, advocating for ethical AI practices, and ensuring compliance with relevant standards. This remote position requires a Bachelor's or Master's degree in a related field and at least 7 years of software engineering experience, including 3 years building and architecting analytical software systems using AI and ML. BambooHR offers a great company culture, comprehensive health insurance, generous leave policies, 401k plans, and a paid vacation bonus.

Requirements

• Bachelor's or Master's degree in Computer Science, Machine Learning, Artificial Intelligence, or related field or equivalent industry experience
• At least 7 years of software engineering experience with at least 3 of those years building and architecting analytical software systems using AI and ML
• Experience developing software utilizing various languages, including Python, Clojure, SQL, and/or the ability to pick up new languages quickly
• Strong knowledge and experience using data platforms, machine learning frameworks, and generative AI tooling such as Databricks, Scikit-learn, TensorFlow, PyTorch, OpenAI / Anthropic LLMs, agentic frameworks
• Experience designing and deploying ML models to production and optimizing MLOps practices
• Experience with the full lifecycle of software and AI/ML development, including version control, build management, unit testing, CI/CD, API paradigms, and model versioning
• Exceptional communication and collaboration skills, including cross-functional teams of data scientists and engineers
• Ability to tactfully and effectively give and receive concrete feedback
• Knowledgeable in responsible AI and security best practices, including OWASP Top 10 and OWASP ML Top 10
• Expertise in deploying and scaling containerized, distributed software and AI systems using tools such as Kubernetes
• Foundational skills in math and statistics that underlie ML
• Depth in using LLMs, including training, fine-tuning, and evaluation. Historical background in “traditional” NLP tools
• Experience in SOA, Modular Monolith Architecture, and distributed systems for AI training and inference

Responsibilities

• Architect, design, and implement scalable AI/ML systems across cloud and on-premise environments
• Design advancements in architecture that move software and AI/ML pipelines forward
• Define architectural changes that can be implemented incrementally, while minimizing risk
• Collaborate with a variety of stakeholders to determine architectural priorities, especially in AI model deployment and MLOps workflows
• Build, fix, and improve code, especially high-value AI/ML services and APIs
• Train team members on AI/ML practices, new techniques, and past mistakes
• Help the team build AI-integrated code with confidence
• Define and evolve model governance, monitoring, drift detection, and re-training workflows
• Advocate for security and ethical AI practices in compliance with OWASP ML Top 10 and relevant standards

Preferred Qualifications

• Familiarity with static analysis, code scanning, and ML-specific monitoring tools
• Familiarity with data engineering and data governance principles

Benefits

• A Great Company Culture that has been recognized by multiple organizations like Inc , and Salt Lake Tribune
• Comprehensive health, life, and disability insurance
• Generous leave policies that include 4 weeks of vacation, 12 company holidays, parental leave, and volunteer time off so you can enjoy quality of life
• 401k plans with up to 6% company match
• $2000 Paid-Paid Vacation bonus
• EAP through Headspace