AI/ML Engineer

We are looking for an enthusiastic and talented Mid-level AI/ML Engineer to join our growing AI/ML team. You will be involved in designing, developing, and optimizing machine learning models, working across various AI domains like NLP, Computer Vision, and Generative AI. In this role, you'll be responsible for implementing solutions, training, and fine-tuning models to deliver real-world impact.

Key Responsibilities

• Design, develop, and deploy machine learning models for a variety of applications, ensuring they meet customer needs.
• Work on training and fine-tuning ML models, with a focus on optimizing performance and scalability.
• Collaborate closely with other engineers and data scientists to build end-to-end machine learning pipelines.
• Assist in evaluating model performance and implement necessary improvements to optimize accuracy and efficiency.
• Contribute to the development of machine learning-based applications for document extraction, conversational AI, and other AI-driven solutions.
• Develop and refine prompts for large language models (LLMs) to ensure they generate accurate and relevant outputs.
• Work with cross-functional teams to preprocess and clean datasets, transforming them into usable features for machine learning models.
• Build and deploy ML pipelines on cloud platforms (e.g., GCP, AWS, or Azure).
• Perform exploratory data analysis (EDA), clean and transform data, and apply techniques to handle data imbalances and outliers.
• Collaborate with senior team members to fine-tune models and integrate them into production environments.

What We’re Looking For

• 3+ years of experience in AI/ML engineering, with hands-on experience in building machine learning models and deploying them into production.
• Solid understanding of machine learning principles, algorithms, and best practices.
• Proficiency in Python and experience with popular libraries such as NumPy, Pandas, Scikit-learn, TensorFlow, and/or PyTorch.
• Familiarity with cloud platforms (e.g., GCP, AWS, or Azure) and their machine learning tools.
• Experience with one or more AI technologies, such as Natural Language Processing (NLP), Computer Vision, or Generative AI (GenAI).
• Understanding of end-to-end machine learning workflows, including data preprocessing, feature engineering, model selection, training, and deployment.
• Knowledge of large language models (LLMs) and experience with prompt development and model fine-tuning.
• Good understanding of working with data, including data cleaning, transformation, and augmentation techniques.
• A passion for learning and staying up to date with the latest developments in AI and machine learning.
• Excellent communication and collaboration skills.

Nice to Have

Familiarity with Google Cloud, AWS, or Azure machine learning services.