Machine Learning Engineer

ML Engineer

Experience: 3-5 years | Location: Bengaluru (Hybrid)

Akaike Technologies is a dynamic and innovative AI-driven company dedicated to building impactful solutions. Our mission is to empower businesses by harnessing the power of data and AI to drive growth, efficiency, and value. We foster a culture of collaboration, creativity, and continuous learning.

Job Description: Focus on PySpark, Databricks & Real-time ML Pipeline

sWe are seeking an experienced and highly skilled
Senior Data Scientis
t to join our team in Bengaluru. This role focuses on driving innovative,
large-scal
e solutions using cutting-edge Classical Machine Learning,
PySpar
k,
Spark SQ
L, and Generative AI. The ideal candidate will possess a blend of deep technical expertise, strong business acumen, effective communication skills, and a sense of ownership. We require a proven track record in designing, developing, and
real-time deploying scalable ML/DL pipeline
s and
LLM Agent
s in a fast-paced, collaborative environment

**.

Key Responsibilities: Emphasizing Data and Deployme**

**nt

Large-Scale Data Handling, PySpark, & Databricks Deployme**

• nt Efficiently handle and mod
  el billions of data poi
  nts using multi-cluster data processing framework
  s (PySp
  ar
  k, Spark
  SQ

**L).

Expertise on Databricks is a must h
ave: Ability to design, write, scale, and moni
tor end-to-end ML Pipel**
ines on Databri

• cks.Proven expertise to run and man
  age Databricks data pipelines in real
  time for low-latency decision-mak
• ing.Design and implement high-performa
  nce API's using Fas
  tAPI in Python to expose real-time and batch ML pipeli

**nes.

ML/DL & GenAI Solution Development & Deplo**

• ymentDesign, implement, and deploy end-to-end ML/DL, GenAI solutions, wri
  ting modular, scalable, and production-ready
• code.Develop and implement scalable deployment pipelines u
  sing D
  ocker and AWS servi
  ces (ECR, Lambda, Step Func
  ti
• ons).Design and implement custom models and loss functions to address data nuances and specific labeling challe
• nges.Apply specialized modeling for marketing scenarios (Targeting, Budget optimisation, Churn) and data limitati
  ons (Sparse/incomplete labels, Single class lea
  rn

**ing).

Generative AI & Large Language Models (LLM A**

• gents)Leverage in-depth understandi
  ng of Transformer archite
  ctures and the principles of Large and Small Language M
• odels.Practical experience in bui
  lding LLM-ready Data Management
  layers for large-scale structured and unstructured
• data.Apply foundational understandi
  ng of LLM
  Agents and multi-agent systems (
  e.g., Agent-Critique,
  ReACT, Agent Collaboration), advanced prompting, LLM evalua
  tion, confidence g
  rading
  , and Human-in-the-Loop s
  y

**stems.

Must-Have Skills: Core Technical Requi**

rementsData Pipelines, PySpark & Data

• bricks Proficiency in
  Python and its data science ecosystem (NumPy, Pandas,
  Dask,
  PySpar
  k) for large-scale data pro
  c
• essing.Expert, hands-on experience with Dat
  abricks for MLOps, pipeline orchestration, and real-time depl
• oyment.Ability to perform effective feature engineering by understanding complex business obje

**ctives.

Core Machine Learning & Deep**

• LearningIn-depth knowledge of ANN, 1
  D/2D/3D Convolutional Neural Networks (C
  onvNets), LST
  Ms, and Transforme
  r
• models.Strong profici
  ency in PU
  le
  arning, single-class
  learni
  ng, and representation
  learning, alongside traditional ML app
• roaches.Advanced understanding and applica
  tion of model explainability te
  chniques (e.g., SHAP
• , LIME).Hands-on experience with ML/DL libraries such as Scikit
  -learn, TensorFlow/Keras, and

**PyTorch.

Gen**

• AI ModelsExperience utilizing large-scale language models (GPT-4, Mistral, Llama, Claude)
  through prompt en
  gineering an
  d custom f
  i

**netuning.

Releva**

• nt to HaveAwareness of best software design practices and backend frameworks l
• ike Flask.Knowledge of Recommender Systems and advanced learning techniques (Representative learning, PU

**learning).

Benefit**

s and PerksCompetitive E

SOP grants.Support for publishing papers and attending academic/industry c

onferences.High visibility across all functions

at Akaike.