Pi Square Technologies - AI/ML Engineer

Job Summary :

We are looking for a highly skilled Machine Learning Engineer to join our growing AI and Data Science team. The ideal candidate will be responsible for designing, developing, deploying, and maintaining machine learning and AI solutions that address complex business challenges. You will work closely with cross-functional teams to build scalable AI systems, optimize model performance, and drive innovation through advanced analytics and emerging AI technologies.

Key Responsibilities :

- Design, develop, and deploy machine learning and AI models aligned with business and client requirements.

- Build end-to-end AI/ML pipelines, including data preprocessing, feature engineering, model training, validation, testing, and deployment.

- Develop and implement solutions across domains such as Natural Language Processing (NLP), Computer Vision, Predictive Analytics, and Generative AI.

- Integrate AI models into enterprise applications through APIs, microservices, and cloud-native architectures.

- Optimize models and inference pipelines for performance, scalability, latency, reliability, and cost efficiency.

- Collaborate with Product Managers, Solution Architects, Data Engineers, and DevOps teams to deliver production-ready AI solutions.

- Implement and maintain MLOps practices, including CI/CD pipelines, model versioning, monitoring, retraining, and governance.

- Monitor model performance in production and continuously improve accuracy, robustness, and operational efficiency.

- Ensure adherence to data privacy regulations, security standards, and Responsible AI principles.

- Contribute to the development of reusable AI frameworks, accelerators, tools, and best practices.

- Stay updated with the latest advancements in machine learning, deep learning, and Generative AI technologies.

Required Skills & Qualifications :

Technical Skills :

- Strong proficiency in Python

- Working knowledge of Java is an added advantage

- Machine Learning & AI Frameworks

- TensorFlow

- PyTorch

- Scikit-learn

- Data Processing & Analytics

- Pandas

- NumPy

- SQL

- Machine Learning Concepts

- Supervised and Unsupervised Learning

- Deep Learning fundamentals

- Model Evaluation and Optimization

- Feature Engineering

- Deployment & Integration

- REST APIs

- Flask and/or FastAPI

- Docker

- Cloud Platforms

- Experience with AWS, Microsoft Azure, or Google Cloud Platform (GCP) for model deployment and pipeline management

- DevOps & Version Control

- Git

- CI/CD tools and workflows

Preferred Qualifications :

- Experience with Generative AI technologies, Large Language Models (LLMs), and Retrieval-Augmented Generation (RAG).

- Exposure to vector databases, prompt engineering, and AI agent frameworks.

- Familiarity with MLOps tools such as MLflow, Kubeflow, SageMaker, or Azure ML.

- Understanding of container orchestration platforms such as Kubernetes.

- Strong analytical, problem-solving, and communication skills.

Education :

UG : Any Graduate (Computer Science, Information Technology, Data Science, Mathematics, Statistics, or related disciplines preferred)