AI Engineer

Nineleaps Technology Solutions Pvt Ltd; Bengaluru; On-site; INR 800,000 - 1,200,000; Full time

Job summary

A technology company specializing in digital engineering is seeking an AI Engineer to build, test, and improve machine learning solutions. The role involves optimizing models using frameworks like TensorFlow and PyTorch, assisting with data collection and analysis, and staying updated with AI developments. Candidates should have strong proficiency in Python, a good grasp of data structures, and familiarity with AI concepts. The position offers a dynamic work environment focused on innovation and collaboration.

Qualifications

• Strong proficiency in Python and hands-on experience with NumPy and pandas.
• Experience with TensorFlow or PyTorch and scikit-learn for machine learning tasks.
• Strong grasp of data structures and algorithms.

Responsibilities

• Implement, test, and optimize machine learning models.
• Assist with data collection, preprocessing, and exploratory data analysis.
• Stay updated with developments in AI and machine learning.

Skills

Python

TensorFlow

PyTorch

scikit-learn

NumPy

pandas

Data structures and algorithms

Git

AI/ML concepts

Data preprocessing

Analytical thinking

Job description

As an AI Engineer at Nineleaps, you will work on building, testing, and improving machine learning solutions that solve real-world business problems. You’ll contribute across the AI lifecycle—from data preparation and model experimentation to evaluation, documentation, and collaboration with cross-functional teams. This role is ideal for professionals with a strong foundation in machine learning and Python who are excited to explore modern AI concepts, including deep learning, LLMs, and generative AI.

Key Responsibilities

• Implement, test, and optimize foundational machine learning models using frameworks such as TensorFlow, PyTorch, or scikit-learn
• Assist with data collection, preprocessing, and exploratory data analysis (EDA) to prepare high-quality datasets for model training and evaluation
• Conduct literature reviews and stay up to date with the latest developments in AI, machine learning, and generative AI
• Apply new ideas and emerging techniques in prototypes and proof-of-concept solutions
• Document technical processes, experiments, and results thoroughly to support reproducibility and knowledge sharing
• Participate actively in code reviews, brainstorming sessions, and team discussions to foster collaboration and innovation
• Present findings, analyses, and model outcomes clearly to both technical and non-technical stakeholders
• Support experimentation and evaluation of models to improve accuracy, efficiency, and business relevance

What We’re Looking For

• Strong proficiency in Python and hands-on experience with core libraries such as NumPy and pandas
• Foundational experience with at least one major deep learning framework, such as TensorFlow or PyTorch, along with scikit-learn for classical machine learning tasks
• Strong grasp of data structures and algorithms (DSA) and core computer science fundamentals
• Familiarity with version control tools such as Git
• Solid understanding of core AI/ML concepts, including supervised learning, unsupervised learning, classification, regression, and model evaluation metrics
• Understanding of deep learning concepts and architectures, especially Transformer models, Large Language Models (LLMs), and generative AI
• Familiarity with data preprocessing, feature selection, and basic data visualization techniques
• Basic conceptual understanding of RAG (Retrieval-Augmented Generation) architectures and prompt engineering techniques
• Strong analytical thinking, attention to detail, and a proactive problem-solving mindset
• Excellent written and verbal communication skills
• Prior internships, academic projects, coursework, or research experience in AI/ML is a strong plus

About Nineleaps

Nineleaps is a technology company specializing in digital engineering, data, and AI. We work with businesses across industries to solve complex challenges through modern technology and create an environment where people can learn fast, take ownership, and build meaningful careers while doing it.