Description

You will design, train, and deploy machine learning models via scalable APIs for real-time and batch inference.

This role is on-site.

Responsibilities

• Design and fine-tune machine learning models using PyTorch, TensorFlow, and Hugging Face Transformers.
• Develop OCR models using Tesseract to extract textual information from documents and images.
• Build and maintain data pipelines for preprocessing, augmentation, and transformation.
• Deploy trained models through RESTful APIs using Flask.
• Optimize model training and inference speeds by leveraging GPU acceleration.

Required Skills

• 5+ years of experience in machine learning model development and training.
• Proficiency in Python for model training, data processing, and API development.
• Hands-on experience training large models on GPUs.
• Deep knowledge of PyTorch, TensorFlow, and Hugging Face Transformers.
• Experience with Tesseract for OCR tasks.
• Strong experience building RESTful APIs with Flask.
• Proficiency with data manipulation libraries including Pandas and NumPy.

Preferred Skills

• Familiarity with Docker for containerization.
• Knowledge of cloud-based solutions such as AWS, GCP, or Azure.

Key Skills

Education

Bachelor’s or Master’s degree in Computer Science, Data Science, Machine Learning, or a related field.