Machine Learning Engineer - Deep Learning (3-7 yrs)

Job Description :

We are hiring ML Engineers / Data Scientists for a AI and Data Engineering Startup in Pune. As a member of the founding team, you will work closely with the GenAI / ML Architect on.

Hands-on Model Development and Deployment :

- Engage in hands-on coding, data preprocessing, feature engineering, model training, and evaluation using frameworks such as TensorFlow, PyTorch, scikit-learn, Hugging Face, and others.

- Develop and optimize CNN architectures for image processing and computer vision tasks, ensuring high accuracy and performance.

- Fine-tune pre-trained LLMs to specific business contexts and requirements, optimizing them for performance, accuracy, and efficiency.

- Deploy AI models in production environments, ensuring scalability, reliability, and low-latency inference, using platforms like AWS SageMaker, Google AI Platform, or Azure ML.

R&D and Innovation :

- Lead research initiatives to explore new algorithms, architectures, and techniques in AI/ML, with a focus on staying ahead of industry trends.

- Innovate and experiment with cutting-edge AI/ML methodologies, including the development of novel RAG strategies for enhancing LLM performance.

- Contribute to the development of proprietary tools and accelerators that can be leveraged across multiple client projects.

Model Evaluation and Continuous Improvement :

- Implement robust model evaluation metrics and frameworks to assess model performance, fairness, and robustness.

- Continuously monitor models in production, making iterative improvements based on real-world performance data and feedback.

- Ensure all AI/ML solutions adhere to ethical AI standards and are free from bias, ensuring fair and responsible AI deployment.

Qualifications :

Experience :

- 2+ years of experience in AI/ML, with a strong background in building and deploying deep learning models, including CNNs for computer vision tasks.

- Extensive hands-on experience with building, fine-tuning, and deploying large language models (LLMs) using frameworks like Hugging Face, GPT, BERT, etc.

- Proven expertise in Natural Language Processing (NLP), including tokenization, embeddings, transformers, and attention mechanisms.

- Experience in developing Retrieval-Augmented Generation (RAG) systems and integrating them into LLM applications.

- Experience with cloud platforms (AWS, Azure, Google Cloud) and MLOps practices, including CI/CD for machine learning.

Technical Skills :

- Proficiency in programming languages such as Python, with deep knowledge of AI/ML libraries like TensorFlow, PyTorch, and scikit-learn.

- Expertise in deep learning techniques, including CNNs for image recognition, and NLP for text processing and understanding.

- Strong understanding of data preprocessing, feature engineering, and model evaluation techniques.

- Experience with databases, data lakes, and data engineering tools to manage and process large-scale datasets.

Soft Skills :

- Strong analytical and problem-solving skills, with the ability to think critically about how to approach and solve complex AI/ML challenges.

- Excellent communication skills, capable of articulating complex technical concepts to both technical and non-technical audiences.

Education :

- Bachelor's or Master's degree in Computer Science, Data Science, Artificial Intelligence, or a related field.

- A Ph.D in a relevant area is a plus.

Why Join Us :

- Be part of a pioneering team driving the next wave of AI and ML innovation.

- Work with senior leadership in the company.

- Work on challenging, high-impact projects that push the boundaries of what AI can achieve.