AI/ML Engineer

About the Role

As an AI/ML Engineer at Driif.ai, you will design, implement, and optimize machine learning models that power real-time decision-making across multi-domain systems. You will work closely with cross-functional teams to understand mission requirements, translate them into ML problem statements, and build robust pipelines for data ingestion, feature engineering, training, evaluation, and deployment. This is a full-time, on-site role based in Bengaluru.

Key Responsibilities

• Researching and experimenting with algorithms.
• Developing and tuning neural network architectures.
• Writing clean and efficient code.
• Conducting performance and scalability testing in production-like environments.
• Contributing to system design, documentation, and best practices for ML operations.
• Ensuring reliability and maintainability of deployed models.

Requirements & Tech Stack

• Strong foundation in Computer Science, including data structures, software engineering principles, and system design.
• Proficiency in Algorithms and Pattern Recognition, with experience framing and solving complex ML problems.
• Solid understanding of Statistics, probability, and experimental design for model evaluation and validation.
• Hands-on experience with Neural Networks, deep learning frameworks (e.g., PyTorch, TensorFlow), and model optimization.
• Experience building end-to-end ML pipelines (data processing, training, deployment) in production environments.
• Strong programming skills in languages commonly used for ML (e.g., Python, C++).
• Familiarity with cloud or edge deployment.
• Ability to work on-site in Bengaluru.
• Ability to collaborate in a fast-paced environment.
• Ability to communicate effectively with technical and non-technical stakeholders.
• Experience with real-time systems, robotics, or mission-critical applications is a plus.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Data Science, Electrical Engineering, or a related quantitative field.