Member of Technical Staff (Machine Learning Engineer)

reka; United States; Remote; USD 140,000 - 210,000; Full time

Benefits offered by this job

5 weeks of paid leave

Healthcare benefits

Visa support

Job summary

Reka is seeking a senior ML engineer to translate research into production-grade machine learning systems. You will design and deploy end-to-end ML models for image and video processing, owning the full lifecycle from experimentation to deployment.

Collaborate with researchers and cross-functional teams to scale ML solutions, leveraging open-source models and cloud platforms. You should have a PhD or MS, deep knowledge of diffusion models and Transformers, and a track record of deploying

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 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 (AWS preferred).

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.

Skills

Research experience

Python

Java/C++/Scala

Diffusion models

CNNs & Transformers

PyTorch or TensorFlow

End-to-end ML deployment

AWS

Education

MS/PhD in CS/EE or related field

Tools

Git

CUDA

Job description

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

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 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 (AWS preferred)

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

Reka's Mission

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.

Why Reka?

• An Elite Team: Collaborate with top-tier engineers, researchers, operators from renowned organizations like Google DeepMind and Facebook AI Research (FAIR) and successful startups, driving innovation in cutting-edge AI technology.

• Massive Market Opportunity: Be part of a rapidly growing industry poised to transform multiple sectors globally, offering the chance to make a significant impact.

• Mission-Driven Environment: Work alongside a collaborative, mission-focused team dedicated to advancing AI for meaningful applications.

• Inclusive and Open Culture: Thrive in an open and inclusive work environment that values diverse perspectives and fosters creativity.

• Generous Benefits: Enjoy 5 weeks of paid leave to recharge, comprehensive healthcare benefits including vision and dental, and additional perks that support your well-being.

• Visa Support: We provide visa assistance, including H1B and OPT transfers, for US employees to ensure a smooth transition and support your career with us.