Key Responsibilities
Research, design, and implement advanced computer vision models for tasks such as image segmentation, pose estimation/transfer, and generative modeling.
Develop effective, scalable code in Python using PyTorch (and other relevant libraries).
Package and deploy ML models as APIs (preferably using FastAPI) for real-world usage.
Build lightweight frontend/backend dashboards or admin panels to monitor, visualize, and manage deployed models.
Collaborate with cross-functional teams (product, data, engineering) to integrate AI solutions into products.
Optimize model performance for inference speed, scalability, and cost efficiency.
Required Skills & Experience
Solid hands-on expertise in Deep Learning and Computer Vision (e.g., segmentation, generative models, pose transfer).
Proficiency in Python and PyTorch for model development.
Experience with API development and model deployment (FastAPI, Flask, or similar).
Familiarity with frontend/backend development for building admin dashboards (React, Vue, Django, or similar).
Solid understanding of software engineering practices (Git, CI/CD, testing).
Experience with GPU-based training and optimization (CUDA, TorchScript, ONNX) is a plus.