Job Description
We are looking for a passionate and skilled AI/ML Engineer with 1–3 years of hands-on experience in computer vision and conversational AI. The ideal candidate should have a strong foundation in machine learning, experience training custom object detection models (especially using YOLO), and working knowledge of OpenCV and AI/ML frameworks. Exposure to OpenAI's ChatGPT and chatbot development is essential. Any experience in AR/VR will be considered a plus.
Key Responsibilities
- Train and deploy custom object detection models using YOLO (v5/v7/v8 or YOLO-NAS) on custom datasets.
- Preprocess and annotate data for training using tools like Roboflow or CVAT.
- Implement and optimize object detection pipelines using OpenCV.
- Develop, fine-tune, and integrate AI chatbots using OpenAI's ChatGPT API or similar LLMs.
- Work with basic ML algorithms for classification, clustering, and regression tasks.
- Collaborate with cross-functional teams to design, test, and deploy AI solutions.
- Explore integration of AI solutions into AR/VR environments (basic knowledge sufficient).
- Maintain and improve model performance through continuous training and validation.
- Stay up-to-date with the latest trends and advancements in AI/ML.
Required Skills
- Strong understanding of YOLO object detection models and custom training pipelines.
- Proficiency in OpenCV for image processing and real-time applications.
- Experience with OpenAI's GPT models and chatbot development.
- Solid understanding of basic ML algorithms (e.g., decision trees, SVM, k-NN).
- Good programming skills in Python.
- Familiarity with AI/ML frameworks such as:
- TensorFlow
- PyTorch
- ONNX Runtime
- Scikit-learn
- Knowledge of tools like Roboflow, LabelImg, or FiftyOne.
- Git version control and experience working in agile environments.
Preferred (Good to Have)
- Exposure to AR/VR technologies (Unity, Unreal, WebXR, etc.).
- Experience in deploying models to edge devices (e.g., Raspberry Pi, NVIDIA Jetson).
Job Details
- Type: Full-time, Permanent
- Location: Mohali, Punjab
- Work Location: In person
- Salary: ₹25,000.00 - ₹30,000.00 per month
- Experience: Minimum 1 year in AI/ML
Benefits
- Flexible schedule
- Paid sick time
Schedule
- Day shift, Monday to Friday, Morning shift
Note
Please ensure to tailor your application highlighting relevant experience and skills in the areas mentioned above.