Machine Learning Engineer

Responsibilities

• 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 to deliver ML solutions at scale
• Participate with advancements in machine learning and apply them to continuously improve products

Requirements

• 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 for low-latency, real-time applications
• Experience deploying ML models in cloud environments (AWS preferred)
• Experience with experiment tracking systems and ML workflows

Preferred Qualifications

• 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

Benefits

• 5 weeks of paid leave
• Comprehensive healthcare benefits including vision and dental
• Visa assistance (H1B and OPT transfers)

About the Company

Reka's mission is to build useful multimodal artificial intelligence and use it to empower organisations and businesses. We are a globally distributed foundation model startup, headquartered in the San Francisco Bay Area, California.