Lead Data Scientist - Computer Vision & OCR

Responsibilities

• Architect, develop, and deploy high-performance Computer Vision and OCR models for real-world applications.
• Implement and optimize state-of-the-art OCR models such as Donut, TrOCR, LayoutLM, and DocFormer for document processing and information extraction.
• Fine-tune and integrate LLMs (GPT, LLaMA, Mistral, etc.) to enhance text understanding and automation.
• Build and optimize end-to-end AI pipelines, ensuring efficient data processing and model deployment.
• Work closely with engineers to operationalize AI models in production using Docker, FastAPI, TensorRT, and ONNX.
• Enhance GPU performance and model inference efficiency through techniques such as quantization and pruning.
• Stay ahead of industry advancements by continuously experimenting with new AI architectures and training techniques.

Requirements

• 5-10 years of experience with proven technical expertise in Python, PyTorch, and TensorFlow.
• Deep experience in Computer Vision and OCR, including developing, training, and deploying document AI models.
• Strong understanding of Transformer-based architectures for vision and text processing.
• Hands-on experience with Hugging Face, OpenCV, TensorRT, and NVIDIA GPUs for model acceleration.
• Proven ability to scale AI solutions and deploy them on cloud platforms (AWS, GCP, or Azure).
• Familiarity with MLOps tools including Docker, FastAPI, and Kubernetes.
• Experience working with multi-modal models (Vision + Text).
• Ability to work autonomously and drive projects from research to production in a fast-paced environment.

Preferred Qualifications

• Strong background in vector databases and RAG pipelines.
• Experience fine-tuning LLMs specifically for document intelligence.
• Contributions to open-source AI projects.

About the Company

Yubi is re-defining global debt markets by freeing the flow of finance between borrowers, lenders, and investors. As a leading fintech unicorn, Yubi provides a holistic product suite for the discovery, investment, fulfillment, and collection of debt solutions. We are a team of over 1,000 motivated individuals working to transform the global institutional debt market through cutting-edge technology.