Design, build, and productionize large-scale machine learning, generative AI, and agentic systems. Own the complete AI product lifecycle from experimentation through production deployment, implementing MLOps workflows with MLflow and Databricks to ensure reliability, scalability, monitoring, and performance. The role requires proven model productionization experience and expertise in Databricks technologies, agent development frameworks, and software repository management.
The ideal candidate possesses deep technical expertise in MLOps and GenAI.
Work Location: Hyderabad, Bangalore, Mumbai, Delhi, Gurgaon.
Work Mode: Work From Office 5 days a week
- End-to-End Development: Design, build, and productionize large-scale machine learning, generative AI and agentic systems and models.
- Operational Excellence: Implement robust MLOps workflows using MLflow to ensure model reliability, scalability, and performance tracking.
- Production Deployment: Take ownership of the full lifecycle of AI products, ensuring models move successfully from experimental stages to live production environments.
- Experience: 4+ years of hands-on experience in Data Science, Machine Learning, GenAI, and MLOps.
- Track Record: Proven experience productionizing models (must have moved models beyond the sandbox/prototype phase).
Technical Stack:
- Required: Expert-level proficiency with MLflow and model deployment and monitoring on Databricks.
- Required: Experience with Databricks Spark, Lakebase, Vector Search.
- Required: Experience with at least one agent development framework like Langchain, Langgraph, DSPy etc.
- Good to have: Hands on experience with Databricks Apps, AI/BI Dashboards, Genie etc.
- Good to have: Databricks Asset Bundle (DAB), Git, experience with large repo management (GitHub, ADO etc.)
Similar Jobs
Information Technology • Software
Lead design, build, and deployment of scalable machine learning models using Python, SQL, and ML/DL frameworks. Process and train models on large datasets with Spark/Databricks/Hadoop, automate pipelines with Airflow and AWS EMR, and integrate models into AWS cloud applications. Drive end-to-end experiments, optimize pipelines and queries, and deliver production-grade, high-impact data science solutions.
Top Skills:
AirflowSparkAws Ec2Aws EmrAws LambdaAws S3Aws SagemakerDatabricksHadoopPandasPythonPyTorchScikit-LearnSQLTensorFlow
Artificial Intelligence • Consumer Web • HR Tech • Other
Design, prototype, and deliver agentic AI solutions using ML and generative models. Partner with product, engineering, and delivery teams to build, evaluate, and deploy LLM-based solutions, run experiments and POCs, and communicate technical recommendations to stakeholders.
Top Skills:
Agentic Ai FrameworksGenerative AiLlmsMachine LearningModel DevelopmentPrompt EngineeringPythonRagSQL
Information Technology
Lead advanced analytics and ML projects: perform EDA, feature engineering, statistical modeling, and build/optimize production ML and Generative AI solutions (LLMs, embeddings, RAG). Collaborate with engineering and MLOps to deploy and monitor models, create visualizations and dashboards, translate insights for stakeholders, and mentor junior team members.
Top Skills:
AWSAzureCi/CdDockerEmbeddingsGCPKubernetesLlmsMlflowMlopsPower BIPrompt EngineeringPythonPyTorchRagScikit-LearnSQLTableauTensorFlowXgboost
What you need to know about the Pune Tech Scene
Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.
