If you're interested, please E-Mail me at chandra.ss@centraprise.com
Position: AI/ML Engineer with Python Exp
Location: Austin, TX/ Plano, TX (Onsite, Hybrid Mode)
Job Type: Fulltime
Job Description:
Responsibilities:
Model Training and Development
O Design, train, and fine-tune machine learning models using frameworks like PyTorch, TensorFlow, and Hugging Face Transformers.
O Utilize GPU resources to optimize model training processes, ensuring high efficiency and low latency.
O Develop OCR models using Tesseract and other tools to extract and process textual information from various types of documents and images.
Data Processing and Preparation
O Conduct thorough data processing and cleaning to ensure high-quality input for training.
O Develop and maintain pipelines for data preprocessing, augmentation, and transformation tailored to specific ML model requirements.
O Collaborate with data engineering teams to ensure seamless integration of data for ML workflows.
API Development and Deployment
O Develop and deploy RESTful APIs with Flask to expose trained ML models and features to downstream applications.
O Build scalable, robust API endpoints to enable real-time and batch model inference.
O Collaborate with DevOps teams to ensure APIs are effectively monitored, scaled, and maintained.
Performance Optimization
O Optimize model training and inference speeds, leveraging GPU acceleration and efficient data processing methods.
O Monitor model performance post-deployment and retrain/update as needed to ensure accuracy and relevance.
Collaboration and Documentation
O Work closely with cross-functional teams, including data scientists, engineers, and product managers, to align ML initiatives with business objectives.
O Document processes, models, and workflows, ensuring reproducibility and knowledge sharing within the team.
Qualifications:
Education: Bachelor's or Master's degree in Computer Science, Data Science, Machine Learning, or a related field.
Experience:
O 3+ years of hands-on experience in machine learning model development and training.
O Proven experience training large models on GPUs, optimizing for both speed and accuracy.
Technical Skills:
O Proficiency in machine learning frameworks like PyTorch, TensorFlow, and Hugging Face Transformers.
O Experience with OCR tools, particularly Tesseract, for text extraction from images.
O Strong experience with Flask for building and deploying RESTful APIs.
O Proficient in Python for data processing, model training, and API development.
O Experience with data manipulation and preprocessing libraries (Pandas, NumPy, etc.).
O Familiarity with Docker for containerizing applications and knowledge of cloud-based solutions (AWS, GCP, or Azure) is a plus.
Other Skills:
O Excellent problem-solving abilities and attention to detail.
O Strong written and verbal communication skills, with an ability to explain complex ML concepts to non-technical stakeholders.