AI Architect

Job Description

About Darwinbox We’re designing the future of work by building the world's best HR tech, driven by a fierce focus on employee experience and customer success, and continuous, iterative innovation. Founded in 2015, we now serve 1.5M+ employees in 650+ enterprises, and that includes massive conglomerates to unicorn startups, like Makemytrip, JSW, Vedanta, Mahindra, and Kotak, plus leading global brands like Nivea, Starbucks, Sephora, AXA, Cigna, T-Systems, and Calvin Klein. We’re backed by marquee investors like TCV, Salesforce, Sequoia, Lightspeed Ventures, and others. Most importantly, we’re growing at a phenomenal pace, and that means exponential growth and learning opportunities for you, plus a band of passionate and fun people to work with within a place where your ideas take precedence over your designation. Why Join Us? The rate at which our product and market presence are growing is unprecedented. We’re a Rocketship. We’re not planning on slowing down anytime soon. And, that’s why we need you! You’ll experience a culture of: Disproportionate Rewards for top performance Accelerated Growth in a hyper-growth environment Wellbeing First culture focused on employee care Continuous Learning and Professional Development Meaningful Relationships and a Collaborative Environment Role Overview We are seeking an experienced AI Architect to lead the development of our AI and Machine Learning infrastructure and specialized language models for HR applications. This role will establish and lead our MLOps practices and drive the creation of scalable, production-ready AI/ML systems. Key Responsibilities Discuss the feasibility of AI/ML use cases along with architectural design with business teams, and translate the vision of business leaders into a realistic technical implementation Play a key role in defining the AI architecture and selecting appropriate technologies from a pool of open-source and commercial offerings Design and implement robust ML infrastructure and deployment pipelines Establish comprehensive MLOps practices for model training, versioning, and deployment Lead the development of HR-specialized language models (SLMs) Implement model monitoring, observability, and performance optimization frameworks Develop and execute fine-tuning strategies for large language models Create and maintain data quality assessment and validation processes Design model versioning systems and A/B testing frameworks Define technical standards and best practices for AI development Optimize infrastructure for cost, performance, and scalability Required Qualifications 7+ years of experience in ML/AI engineering or related technical roles 3+ years of hands-on experience with MLOps and production ML systems Demonstrated expertise in fine-tuning and adapting foundation models Strong knowledge of model serving infrastructure and orchestration Proficiency with MLOps tools (MLflow, Kubeflow, Weights & Biases, etc.) Experience implementing model versioning and A/B testing frameworks Strong background in data quality methodologies for ML training Proficiency in Python and ML frameworks (PyTorch, TensorFlow, Hugging Face) Experience with cloud-based ML platforms (AWS, Azure, Google Cloud) Proven track record of deploying ML models at scale Preferred Qualifications Experience developing AI applications for enterprise software domains Knowledge of distributed training techniques and infrastructure Experience with retrieval-augmented generation (RAG) systems Familiarity with vector databases (Pinecone, Weaviate, Milvus) Understanding of responsible AI practices and bias mitigation Bachelor's or Master's degree in Computer Science, Machine Learning, or related field What We Offer Opportunity to shape AI strategy for a fast-growing HR technology leader Collaborative environment focused on innovation and impact Competitive compensation package Professional development opportunities Flexible work arrangements Qualified candidates who are passionate about applying cutting-edge AI to transform HR technology are encouraged to apply.