AI/ML Engineer with Python Exp (US Citizen, Green card)

If you're interested, please E-Mail me at chandra.sscentraprise.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. Thanks & Regards, Chandra Shekar | Centraprise Corp