Back to jobs

ML-Ops

Sia
Mumbai, Maharashtra, India (Hybrid)
Full-time
AI tools:
scikit-learn
You apply on Sia's own careers site

This hands-on role maintains and improves production algorithms, with a focus on Python performance, concurrency, testing, and REST APIs. It also deploys machine-learning models as services and works with product, data, and QA teams to meet reliability and scalability goals.

Full-time
On-site
5–7 years

Skills & Expertise

Python
Django/DRF
FastAPI
scikit-learn
pandas
NumPy
Profiling tools (cProfile, line_profiler
pytest

Key Responsibilities

Maintain production algorithms, fix bugs, and add regression tests.

Profile and optimize code, applying parallelism or concurrency where useful.

Build and maintain REST APIs and test service efficiency and scalability.

Full Description

Job Description

We’re looking for a hands-on Data Scientist with strong Python and backend skills to maintain and improve production algorithms. The focus is on performance optimization, parallel processing, bug fixing, and building/maintaining REST APIs (Django/FastAPI). ML/DL is applied pragmatically; GCP is a plus.

Key Responsibilities

• Maintain and improve production algorithms; triage and fix bugs; add regression tests

• Profile and optimize code (CPU/memory/I/O); apply parallelism/concurrency where relevant

• Design and maintain REST APIs (Django/DRF or FastAPI) for model/algorithm services

• Test for efficiency and scalability; benchmark latency/throughput; right-size costs

• Ensure high code quality: unit/integration tests, type hints, CI/CD, documentation

• Collaborate with product/data/QA to prioritize improvements and meet SLAs

Qualifications

• 5–7 years in data science/software roles with production-grade Python

• Strong Python expertise (pandas, NumPy), profiling tools (cProfile, line_profiler, memory_profiler, py-spy)

• Parallel/concurrent processing (multiprocessing, threading, asyncio) and performance tuning

• REST API development with Django/DRF or FastAPI (auth, error handling, pagination)

• ML fundamentals with scikit-learn; experience deploying models to services

• Testing/quality: pytest/unit test, CI/CD, Git workflows; clear communication and documentation

Additional Information

What We Offer

• Opportunity to lead cutting-edge AI projects in a global consulting environment.

• Leadership development programs and training sessions at our global centers.

• A dynamic and collaborative team environment with diverse projects.

Position based in Mumbai (onsite)

Sia is an equal opportunity employer. All aspects of employment, including hiring, promotion, remuneration, or discipline, are based solely on performance, competence, conduct, or business needs.

Applications are handled on Sia's site