AI/ML Engineer - I

Geekyants; Bengaluru; On-site; INR 1,200,000 - 1,800,000; Full time

Job summary

Geekyants is seeking an AI/ML Engineer I to design, develop, and deploy machine learning models and AI solutions in Bengaluru. You will work with modern ML frameworks, large language models, and build scalable AI applications.

You will develop ML pipelines, leverage LLMs with RAG systems, and deliver AI-driven services with REST APIs using FastAPI or Flask, deployed via Docker and Kubernetes. Strong Python skills and ML tooling are essential for success in this role.

Qualifications

• Bachelor's or Master's degree in Computer Science, Data Science, or a related field.
• Approximately 1 year of experience in ML/AI development.
• Familiarity with MLOps tools like MLflow or Weights & Biases.
• Knowledge of computer vision and/or NLP applications.

Responsibilities

• Design, develop, and deploy machine learning models and AI solutions.
• Build end-to-end ML pipelines from data preprocessing to model serving.
• Work with LLMs, implement RAG systems, and develop AI-driven applications.
• Create REST APIs using FastAPI or Flask and deploy using Docker and Kubernetes.
• Process and analyze large datasets using SQL, Pandas, and Spark.
• Implement MLOps practices such as model monitoring, logging, and versioning.

Skills

Python

Software Architecture

Version Control

ML/AI Technologies

LLM Expertise

Vector Databases

Data & Infrastructure

Deployment

Mathematical Foundation

Soft Skills

Education

Bachelor's or Master's in CS/Data Science or related

Tools

TensorFlow

PyTorch

Scikit-Learn

XGBoost

FastAPI

Flask

Docker

Kubernetes

SQL

Pandas

NumPy

Apache Spark

AWS

GCP

Azure

MLflow

Weights & Biases

Job description

We are seeking an AI/ML Engineer - I to design, develop, and deploy machine learning models and AI solutions. This role involves working with modern ML frameworks, large language models (LLMs), and building scalable AI applications.

Key Responsibilities

• Develop and deploy machine learning models using TensorFlow, PyTorch, and Scikit-Learn
• Build end-to-end ML pipelines from data preprocessing to model serving
• Work with LLMs, implement RAG systems, and develop AI-driven applications
• Create REST APIs using FastAPI or Flask and deploy using Docker and Kubernetes
• Process and analyze large datasets using SQL, Pandas, and Spark
• Implement MLOps practices such as model monitoring, logging, and versioning

Required Skills

• Programming & Development
• Python - Strong proficiency, including OOP principles
• Software Architecture - Familiarity with frontend/backend components, microservices, and API design
• Version Control - Proficient with Git and platforms like GitHub or GitLab
• ML/AI Technologies
• ML Frameworks - TensorFlow, PyTorch, Scikit-Learn, XGBoost
• LLM Expertise - Understanding of transformer architectures, fine-tuning, and prompt engineering
• Vector Databases - Experience with Pinecone, Weaviate, or Chroma for embeddings and similarity search
• Data & Infrastructure
• Data Handling - Pandas, NumPy, SQL, Apache Spark
• Cloud Platforms - Experience with AWS, GCP, or Azure
• Deployment - Docker, Kubernetes, FastAPI, Flask, REST APIs
• Mathematical Foundation
• Solid understanding of linear algebra, statistics, and calculus as applied to ML algorithms
• Soft Skills
• Strong problem-solving abilities, clear technical communication, and collaboration across teams

Preferred Qualifications

• Bachelor's or Master's degree in Computer Science, Data Science, or a related field
• Approximately 1 year of experience in ML/AI development
• Familiarity with MLOps tools like MLflow or Weights & Biases
• Knowledge of computer vision and or NLP applications

Additional Technical Areas

• Understanding of data privacy and security best practices
• Experience with streaming data processing tools such as Kafka or Apache Beam