Artificial Intelligence/Machine Learning Engineer

Job Title: AI/ML Developer

Location: Sahibzada Ajit Singh Nagar, Punjab, India | Job Type: Full-Time

Summary:

We are seeking an experienced AI/ML professional with strong Python skills, Django/Flask expertise, and hands-on experience in deploying applications on AWS. The ideal candidate will also be proficient in Django Rest Framework (DRF), ML frameworks like PyTorch and TensorFlow, and transformer architectures such as BERT or GPT.

Key Responsibilities:

• Develop web apps with Django & Flask
• Build RESTful APIs using DRF
• Deploy & manage on AWS (EC2, S3, RDS, Lambda, CloudFormation)
• Integrate AI/ML APIs into systems
• Build ML models (PyTorch, TensorFlow, scikit-learn)
• Implement transformers for NLP (e.g., BERT, GPT)
• Optimize models (tuning, pruning, quantization)
• Deploy models via TensorFlow Serving, Torch Serve, and Sage Maker
• Ensure performance, scalability, and reliability
• Collaborate across teams & deliver solutions
• Write clean, efficient, maintainable code
• Conduct code reviews
• Stay updated on AI/ML trends

Qualifications:

• Bachelor’s in CS, Engineering, or related field
• Strong Python skills
• Django & Flask experience
• AWS deployment knowledge
• Proficient in DRF, PyTorch, TensorFlow, scikit-learn
• Experience with transformers for NLP
• SQL/NoSQL (PostgreSQL, MongoDB) knowledge
• Familiar with MLOps practices
• Basic front-end tech knowledge (HTML, CSS, JS)
• Excellent problem-solving & communication skills