Python Ai/ML Engineer

Job Description

Profile: Senior Python AI/ML Engineer

Required experience: 5-9 Years

Location: Mohali, Punjab (WFO only, no hybrid )

B.tech/ MCA Preferred

Immediate joiners

We are looking for a Senior Python (AI/ML) Engineerto build and scale a production-grade AI-powered platform focused on image understanding, scene reconstruction, and automated media generation. This is a backend-heavy role requiring strong expertise in system design, asynchronous processing, and scalable API development.

You will work closely with AI/ML, frontend, and DevOps teams to deliver high-performance systems that power object detection, segmentation, and 3D reconstruction pipelines.

Key Responsibilities

• Design and develop scalable backend services and APIs using FastAPI
• Build modular,maintainable systems with clear service contracts and abstractions
• Implement asynchronous processing pipelines for compute-heavy AI workloads
• Design and manage job queues (Celery, RQ, or similar) and integrate Redis
• Handle long-running and GPU-based tasks- Work with PostgreSQL and integrate object storage (S3 or equivalent)
• Integrate and productionize ML models
• Optimize inference pipelines for latency, batching, and throughput
• Ensure system reliability with retries, fallbacks, and monitoring
• Collaborate with Data Science, Frontend, and DevOps teams

Required Skills

• 5+ years of Python development experience
• Hands-on experience with FastAPI and REST APIs
• Strong understanding of async programming and scalable systems
• Experience with Redis and queue systems (Celery, RQ, Kafka)
• Strong PostgreSQL experience
• Experience with Docker (mandatory); Kubernetes is a plus
• Familiarity with AWS, GCP, or Azure

Good To Have

• Exposure to AI/ML pipelines (YOLO, DETR, SAM, etc.)
• Experience deploying ML models in production
• Understanding of GPU workloads and optimization
• Background in image/video processing or real-time systems

What We’re Looking For

• Strong problem-solving and system design mindset
• Ability to handle real-world scale and unstructured data
• Experience in AI-integrated backend systems

Benefits

• Flexible schedule
• Health insurance
• Leave encashment
• Provident Fund

Skills:- FastAPI, Celery, Redis, PostgreSQL, RESTful APIs, Apache Kafka, Docker and Kubernetes