Member of Technical Staff (Machine Learning Engineer)

Stack

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 (product, engineering, design)

Requirements

• ▹MS/PhD in Computer Science, Electrical Engineering, or related field
• ▹Strong research experience with familiarity in top conferences (CVPR, ICCV, NeurIPS)
• ▹5+ years of Python experience, proficiency in Java, C++, or Scala
• ▹Strong understanding of diffusion models
• ▹Solid knowledge of multi-threading and memory management
• ▹Knowledge of ML architectures: CNNs and Transformers
• ▹Experience with PyTorch or TensorFlow
• ▹Experience building end-to-end, low-latency ML deployment and inference systems
• ▹Experience deploying ML models in cloud environments (AWS preferred)
• ▹Experience with experiment tracking systems and ML workflows

Nice to have

• ▹Experience in low-level optimization (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 with tools like Spark

Soft skills

What we offer

• ▹Elite team: collaborate with alumni from Google DeepMind and Facebook AI Research (FAIR)
• ▹Massive market opportunity in a rapidly growing industry
• ▹Mission-driven, collaborative environment
• ▹Inclusive and open culture
• ▹5 weeks of paid leave
• ▹Comprehensive healthcare including vision and dental
• ▹Visa support (H1B, OPT transfer) for US employees

About the company

Reka's mission is to build useful multimodal artificial intelligence to empower organizations and businesses. A globally distributed, remote-first foundation model startup headquartered in the San Francisco Bay Area, its team includes talent from around the world, including alumni of Google DeepMind and Facebook AI Research (FAIR).