Member of Technical Staff (Machine Learning Engineer)

The role

This person will translate research into production-ready machine learning systems. They will design, build, and deploy end-to-end ML models and pipelines for image and video processing.

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

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

Benefits

• 5 weeks of paid leave
• comprehensive healthcare benefits including vision and dental
• additional perks that support your well-being

Before you apply

• We provide visa assistance, including H1B and OPT transfers, for US employees

About the company

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