Member of Technical Staff: Machine Learning Engineer

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

• 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 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

• Experience in low level optimisation, cuda etc.
• Experience product ionising 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

• Generous paid leave: 5 weeks
• Comprehensive healthcare benefits including vision and dental
• Visa support (H1B and OPT transfers) for US employees

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. Embracing a remote‑first approach, our team brings together top talent from around the world. Our founding team, along with many of our team members, has contributed to many of the breakthroughs in AI over the past decade.