AI/ML Engineer (Python / TensorFlow)

Tech Stack

Key Responsibilities

• Design and implement ML models for NLP, computer vision, and predictive analytics
• Build and maintain data pipelines for model training and inference
• Integrate AI models into enterprise applications via APIs
• Conduct experiments, analyse results, and iterate on model performance
• Work closely with data engineers and software developers
• Document models, experiments, and deployment processes

Requirements

• 4-7 years of experience in machine learning / AI engineering
• Strong Python programming skills (NumPy, Pandas, Scikit-learn)
• Hands-on experience with TensorFlow or PyTorch
• Experience deploying ML models to production (MLOps)
• Knowledge of LLM fine-tuning and prompt engineering
• Strong understanding of statistics and mathematics

Nice to Have

• Experience with LangChain or LlamaIndex
• Knowledge of vector databases (Pinecone, Weaviate)
• Experience with OpenAI / Azure OpenAI APIs
• Cloud ML platforms (AWS SageMaker, Azure ML, Vertex AI)

What We Offer