Job Description Position: AI EngineerTeam:AI Transformations TeamLocation:Gurgaon About Tata 1mgTata 1mg is a trusted health partner for end consumers and large institutions. Our Mission is to make healthcare understandable, accessible and affordable for all Indians. Over the past three years, we've worked to build a healthcare platform that guides customers to the right and affordable care. Today, Tata 1mg is present in 1000+ cities- And we're just getting started. Leading the chart as one of the top consumer health platforms through ePharmacy, eDiagnostics, e-consult and offline stores.Tata 1mg has delivered over 15M ePharmacy orders, 2M eDiagnostics orders & 1M+ organic users/day in 2022 and we are just getting started. Our strongest asset is our 6400+ people enabling us to deliver on our mission consistently.Know more about us:https://www.1mg.com/aboutUsBrief about the TeamThe AI Transformations team at Tata 1mg leads innovation in artificial intelligence and machine learning to drive healthcare transformation. This team works on strategic, high-impact AI projects across multiple business verticals that aim to deliver high impact. What will you do:Model Development: Develop, train, and validate ML models (LLMs and other deep neural networks) for a variety of healthcare use-cases, from image analysis to time series forecasting. You will experiment with SOTA architectures using frameworks like HugginFace, PyTorch, Dart etcModel Deployment: Deploy ML models as scalable APIs or microservices. Leverage tools like FastAPI to build RESTful endpoints and Docker to containerize applications, ensuring models are robust and performant in production environments.Data/Error Analysis & Validation: Analyze large datasets (e.g. Inventory/Sales data, Image datasets) to extract patterns and insights that inform model improvements. Validate model outputs against real-world data and ensure accuracy, reliability of AI solutions.Collaboration & Reporting: Collaborate with cross-functional teams - from data engineers to business - to gather requirements and deploy solutions. Communicate your findings and model performance through clear documentation, presentations, and dashboards to drive data-driven decision making.Vendor Engagement & Collaboration: Work closely with external technology and data vendors to align on solution design, integration, and delivery timelines. Ensure their solutions meet business and technical requirements, and provide feedback or guidance to improve quality and efficiency of joint initiatives.What are we Looking for:Programming: Proficiency in Python and SQL for data analysis and machine learning. Familiarity with libraries like pandas, NumPy, and scikit-learn.Machine Learning Frameworks: Hands-on experience with PyTorch, scikit-learn, and transformer-based models.Transformers and other deep architectures: Experience working with Hugging Face for NLP tasks or LLM fine-tuning/inference.API Development: Basic knowledge of web frameworks (e.g. FastAPI or Flask) to create endpoints for model inference.DevOps & Deployment: Comfort with containerization and deployment tools. Experience using Docker to package applications, and understanding of CI/CD pipelines or cloud platforms is advantageous.Knowledge of time series, image classification, or text processing techniques is a plus.Soft Skills: Strong analytical and problem-solving abilities, good communication skills, and the ability to work in a team-oriented, agile environment. A curious mindset with a passion for staying up-to-date with the latest AI research.Why Join Us?Tata 1mg provides one of the most exciting and high-growth environments to work. You get the added benefit of working in a prestigious legacy brand like Tata and a highly dynamic and versatile startup like 1mg. Being an integral part of the Business Intelligence will provide a great opportunity to scale up a business to great heights leveraging the existing capabilities of Tata 1mg.Disclaimer: Tata 1mg is proud to be an Equal Opportunity Employer. All employment decisions at Tata 1mg shall be made without discrimination, harassment or less favorable treatment of any employee or job applicant, either directly or indirectly, on the grounds of age, color, physical ability, ethnic origin, nationality, religion, gender, family status, marital status, prenatal status, gender reassignment, or sexual orientation. All employment decisions are based on business needs, job requirements, competence and merit.