Member of Technical Staff (Machine Learning Engineer) at Reka AI

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 Py Torch or Tensor Flow

• 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