Python Developer (AI Experience)

About Us:

We are an innovation-driven software company developing an AI-powered wedding planner and virtual try-on platform. Our mission is to combine artificial intelligence, computer vision, and intelligent automation to create next-generation digital experiences.
We are looking for an experienced Python Developer who possesses deep knowledge of Python libraries, AI model integration, and scalable backend design.

Job Summary:

The ideal candidate is a Python expert with extensive experience in AI model development, API creation, and data-driven systems. You will work on building, integrating, and optimizing AI/ML pipelines, handling computer vision tasks, and designing high-performance APIs that connect with LLMs and generative AI models.

Key Responsibilities:

• Design, develop, and optimize AI-driven backend services using Python.
• Work extensively with AI/ML libraries such as PyTorch, TensorFlow, Transformers, OpenCV, NumPy, Pandas, and Scikit-learn.
• Build and deploy LLMs, vision models, and diffusion-based models for virtual try-on and planner modules.
• Develop RESTful / GraphQL APIs and integrate them with AI/ML inference pipelines.
• Manage and optimize databases using PostgreSQL / MongoDB.
• Integrate third-party AI SDKs and APIs (e.g., Hugging Face, Vertex AI, OpenAI, Stability AI).
• Work on asynchronous processing, real-time data flows, and task queues (Celery, Redis, Kafka).
• Implement model versioning, performance tuning, and GPU optimization.
• Collaborate with ML engineers, frontend teams, and DevOps for full lifecycle deployment.
• Conduct research on the latest AI frameworks, libraries, and model fine-tuning techniques.

Required Technical Skills:

• Expert-level proficiency in Python (3.8+) and OOP concepts.
• Strong hands-on experience with AI/ML frameworks:
• PyTorch, TensorFlow, Keras, Transformers, LangChain
• OpenCV, Scikit-learn, NumPy, Pandas, Matplotlib, Seaborn
• Proven experience deploying LLMs, diffusion models, or computer vision models.
• Strong understanding of API development, microservices architecture, and backend optimization.
• Experience with Docker, Git, FastAPI/Flask, and API documentation tools (Swagger, Postman).
• Familiarity with GPU-based model inference, CUDA optimization, and cloud services (GCP, AWS, Azure).
• Solid grasp of data preprocessing, model training, and fine-tuning workflows.
• Knowledge of CI/CD pipelines and model deployment best practices.

Bonus Skills:

• Experience with LangChain, LLM orchestration, and vector databases (FAISS, Pinecone).
• Understanding of Stable Diffusion, ControlNet, Segment Anything, or image generation models.
• Experience with cloud-based AI tools like Vertex AI, OpenAI API, or Hugging Face Hub.
• Exposure to prompt engineering, data labeling, and automated inference workflows.

Qualifications:

• B.Tech / M.Tech / MCA in Computer Science, AI/ML, or related field.
• 1.5–4 years of professional experience in Python and AI model integration.
• Proven track record in delivering production-grade AI systems or backend architectures.

Perks & Benefits:

• Work on cutting-edge AI applications in the wedding & fashion domain
• Flexible work hours & 6-day work week
• Continuous learning & innovation-driven environment
• Opportunity to lead AI-driven product development

Job Type: Full-time

Pay: ₹10,078.22 - ₹30,878.10 per month

Work Location: In person