AI Engineer

Overview

Assist in developing, training and deploying machine learning and deep learning models, including NLP and LLM applications, at YipitData's fully remote India-based team.

What you'll do

• Assist in the development, training, and fine-tuning of machine learning (ML) models.
• Work with large datasets, including data cleaning, preprocessing, feature engineering, and preparing high-quality training data.
• Implement and optimize deep learning models using frameworks such as TensorFlow, PyTorch, or Scikit-learn.
• Support research and development of AI applications in natural language processing (NLP), large language models (LLMs), and predictive analytics.
• Collaborate with software engineers to deploy and integrate AI models into production environments, including model-integrated API services.
• Conduct experiments, analyze results, and enhance model accuracy, efficiency, latency, and performance.
• Stay updated with the latest advancements in AI and machine learning.

What you'll need

• Bachelor's or Master's degree in Computer Science, AI, Data Science, or a related field.
• 3 to 4 years of relevant hands-on work experience.
• Solid understanding of machine learning algorithms and deep learning models.
• Hands-on experience with Python and AI frameworks (TensorFlow or PyTorch).
• Familiarity with data processing libraries like Pandas, NumPy, and regular expressions.
• Understanding of model evaluation techniques and performance metrics.
• Experience with model training, fine-tuning, and experimentation.
• Familiarity with NLP and LLM applications.

Nice to have

• Experience with NLP and LLM frameworks (Hugging Face Transformers, spaCy).
• Exposure to cloud AI services (AWS SageMaker, Google Vertex AI, Azure AI).
• Knowledge of MLOps practices, including CI/CD for ML models.
• Familiarity with AI ethics, bias detection, and model interpretability.

Details

• Fully-remote opportunity based in India.
• Standard work hours are from 8 am to 5 pm IST.