Job Description
What You’ll Do
Translate cutting-edge research into production-ready machine learning systems
Design, build, and deploy end-to-end ML models and pipelines
Develop and optimize models for image and video processing
Own the full ML lifecycle: experimentation, training/fine-tuning, evaluation, and deployment
Rapidly prototype using open-source models and adapt them for product needs
Conduct experiments, analyze results, and iterate to improve performance
Collaborate with researchers and cross-functional teams (product, engineering, design) to deliver ML solutions at scale
Participate with advancements in machine learning and apply them to continuously improve products
What We’re Looking For
Required Qualifications
MS/PhD in Computer Science, Electrical Engineering, or related field
Strong research experience with familiarity in top conferences (e.g., CVPR, ICCV, NeurIPS)
5+ years of experience in Python and proficiency in Java, C++, or Scala
Strong understanding of diffusion models
Strong understanding of multi-threading and memory management
Solid knowledge of ML architectures: CNNs and Transformers
Experience with PyTorch or TensorFlow
Experience building end-to-end ML deployment and inference systems, especially for low-latency, real-time applications
Experience deploying ML models in cloud environments (AWS preferred)
Experience with experiment tracking systems and ML workflows
Nice to Have
Experience in low level optimisation, cuda etc.
Experience productionizing and scaling ML models in real-world systems
Contributions to open-source projects
Experience with MLOps tools or distributed training systems
Familiarity with relational databases (Postgres/MySQL)
Experience handling large-scale data using tools like Spark
