Member of Technical Staff (ML)

Reka; United States; Remote; USD 150,000 - 230,000; Full time

Benefits offered by this job

Visa support

Job summary

Reka, a globally distributed foundation model startup headquartered in the San Francisco Bay Area with a remote-first approach, is seeking a Member of Technical Staff on Machine Learning. You will contribute to the full development cycle of our cutting-edge large deep learning models.

You will prepare datasets, design architectures, implement solutions, and train and evaluate models to improve our products, collaborating closely with engineers and researchers to translate research into practical

Qualifications

• Experience with training and evaluating large deep learning models.
• Proficiency in PyTorch and/or JAX.
• Proven track record of applying ML algorithms at scale in software.
• Ability to work in a fast-paced environment with some uncertainty.
• Collaborative and kind work style.

Responsibilities

• Contribute to the full development cycle of cutting-edge large DL models.
• Prepare datasets, design architectures, implement solutions, train and evaluate models.
• Collaborate with engineers and researchers to translate research into practical applications.
• Join at an exciting moment, wear many hats, help build from the ground up.

Skills

Large DL models

ML frameworks: PyTorch, JAX

ML algorithms at scale

Fast-paced environment

Collaborative work style

Tools

PyTorch

JAX

Job description

As a Member of Technical Staff on Machine Learning, you will:

• Contribute to the entire development cycle of our cutting-edge large deep learning models.
• Prepare datasets, design architectures, implement solutions, train and evaluate models to improve our products.
• Collaborate closely with engineers and researchers to translate cutting-edge research into practical applications.
• Join us at an exciting moment, wear many hats and help us build something from the ground up!

You may be a good fit, if you have:

• Experience with training and evaluating large deep learning models.
• Proficiency in standard deep learning frameworks (e.g. PyTorch, JAX).
• Proven track record of applying and implementing machine learning algorithms into software applications at scale.
• The ability to build in a fast-paced environment under some uncertainty.
• A kind and collaborative nature and work style.

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.