Software Engineer, Machine Learning

Role Overview

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.

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.

Requirements

• 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.

Skills

• Python
• PyTorch
• TensorFlow
• Generative AI
• OpenCV

Benefits

• Health insurance
• Provident Fund