Software Engineer - Machine Learning

Job Description:

We are looking for a highly skilled and driven AI Engineer to join our AI/ML engineering team. The ideal candidate must have strong hands-on experience with Python and deep learning frameworks, an excellent understanding of neural network architectures, and expertise in generative AI. In addition to advanced model design and optimization, the candidate should have working knowledge of anomaly detection, computer vision, image processing, and GPU-based acceleration techniques.

Key Responsibilities:

• Design, develop, and deploy AI models using Python, TensorFlow, PyTorch, and Keras.
• Implement advanced deep learning architectures including CNNs, RNNs, GANs, and transformer-based models.
• Apply transfer learning and fine-tuning for computer vision and anomaly detection tasks.
• Build and manage distributed training pipelines using multi-GPU or multi-node setups.
• Integrate Generative AI models (e.g., Stable Diffusion, GANs, LLMs) for various real-world applications.
• Work on object detection, image segmentation, and image processing tasks using OpenCV or related libraries.
• Analyze complex datasets and present actionable insights using visualizations and reports.
• Leverage GPU computing (CUDA, TensorRT, cuDNN) to optimize training and inference workloads.
• Contribute to model deployment and inference optimization for production environments.
• Develop backend APIs and tools using Django for AI model integration and monitoring.

Required Skills & Qualifications:

• Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, or a related discipline.
• Strong hands-on programming experience in Python.
• Practical knowledge of deep learning frameworks: TensorFlow, PyTorch, Keras.
• Familiarity with both Windows and Linux operating systems.

Job Type: Full-time

Pay: ₹25,000.00 - ₹45,000.00 per month

Benefits:

• Health insurance
• Paid sick time
• Provident Fund

Experience:

• total work: 1 year (Preferred)

Work Location: In person