Truminds - AI/ML Software Lead

Description :

We are looking for an experienced AI/ML Software Lead / Expert to drive software development initiatives and lead the design and implementation of cutting-edge AI/ML solutions. This hybrid role, based in Bengaluru South, involves working closely with cross-functional teams to build high-quality, scalable, and efficient AI-enabled software products.

The role requires strong leadership, expertise in embedded software development, AI/ML engineering, testing, optimization, and deep involvement in delivering end-to-end software solutions that meet complex client requirements.

Key Responsibilities :

AI/ML Solution Design & Development :

- Lead the development of AI/ML models, algorithms, and predictive solutions.

- Design and optimize end-to-end AI pipelines, including data preprocessing, model development, evaluation, and deployment.

- Work with frameworks such as TensorFlow, PyTorch, Keras, Scikit-learn, or similar AI ecosystems.

- Collaborate with data scientists and domain experts to translate business needs into intelligent software features.

Software Engineering & Embedded Development :

- Oversee embedded software development, firmware programming, and integration of AI/ML models into constrained hardware systems.

- Drive software architecture decisions ensuring robustness, maintainability, and scalability.

- Optimize code for performance, memory, and real-time execution on embedded platforms.

Testing, Validation & Optimization :

- Define and implement unit tests, integration tests, and system-level validation for embedded and AI/ML components.

- Conduct debugging and performance tuning to ensure smooth functioning of all modules.

- Ensure high-quality releases through continuous integration and test automation.

Technical Leadership :

- Lead software development teams, mentor engineers, and review code/design deliverables.

- Drive technical decision-making, architectural improvements, and best practices adoption.

- Coordinate with cross-functional teams including hardware, QA, product, and AI/ML teams.

Collaboration & Process Excellence :

- Work within Agile development methodologies, contributing to sprint planning, retrospectives, and improvement cycles.

- Maintain accurate documentation across design specs, model documentation, test cases, and release notes.

- Communicate effectively with internal stakeholders and external clients.

Required Qualifications & Skills :

Core Technical Skills :

- Strong proficiency in Python, C/C++, Java, or similar languages.

- Solid understanding of AI/ML fundamentals including :

1. Deep Learning (CNNs, RNNs, Transformers)

2. Machine Learning algorithms (Regression, SVM, clustering, decision trees, etc.)

3. Computer Vision (OpenCV, image processing, object detection)

4. NLP techniques (tokenization, embeddings, transformers)

5. Model optimization (quantization, pruning, acceleration)

- Experience with AI/ML frameworks & tools :

1. TensorFlow, PyTorch, Keras, ONNX, Scikit-learn

2. OpenVINO / TensorRT (for embedded inference)

- Hands-on experience in embedded systems, RTOS, microcontrollers, or edge AI deployments.

Additional Skills :

- Strong background in Software Architecture and design patterns.

- Experience with testing frameworks, debugging tools, and profiling methodologies.

- Familiarity with Docker, Git, CI/CD pipelines, and modern DevOps tools.

- Understanding of Agile methodologies and complete SDLC processes.

Professional & Educational Background :

- Bachelors or Masters degree in Computer Science, Engineering, Electronics, or related fields.

- Prior experience in AI/ML solution development is highly desirable.

- Excellent communication, analytical, and problem-solving abilities.