Research Engineer / Research Scientist

River AI; Palo Alto (CA); On-site; USD 200,000 - 420,000; Full time

Benefits offered by this job

Health, dental & vision insurance

Unlimited PTO

relocation assistance

Job summary

River seeks exceptional researchers to design and train foundation models powering its personal AI. You will push the frontier of deep learning, focusing on architectures that learn continuously, personalize deeply, and run efficiently on local hardware.

You will own the research lifecycle from ideation to scaling large training runs, ensuring AI evolves with users as a true extension of their will. Compensation reflects high performance and impact.

Qualifications

• BS, MS or Ph.D. in CS, ML, Mathematics or equivalent practical industry experience.
• Deep understanding of modern DL architectures (Transformers, diffusion) and training.
• Hands-on with PyTorch or JAX and scalable model training code.
• Ability to take open-ended research problems from theory to scalable implementations.
• Collaborative mindset with bias for action in a fast-paced environment.

Responsibilities

• Design, train, and evaluate foundation models for personalization and local inference.
• Advance continual learning and on-device adaptation without forgetting.
• Develop data-efficient, RLHF-aligned, PEFT techniques for bespoke hardware.
• Collaborate with infrastructure to scale experiments on large compute clusters.

Skills

Deep learning architectures

PyTorch/JAX

Research to implementation

Collaborative mindset

Edge inference

Education

BS/MS/PhD in CS/ML/Math

Job description

At River, our mission is to create personal AI owned and shaped by each individual. To achieve this, we are rewriting the entire stack from scratch: personal hardware for local inference, bespoke training infrastructure, next-generation UIs, and frontier deep learning research.

Who we are

We are scientists, engineers, and builders from the industry's top tech companies and AI labs. We bring a proven track record of scaling consumer systems for hundreds of millions of users and architecting the pre-training infrastructure behind today's frontier models.

About the Role

We are looking for exceptional researchers to design and train the foundation models that power River's personal AI. Your goal is to push the frontier of deep learning, focusing on architectures that can continuously learn, deeply personalize, and run efficiently on local hardware.

You will take ownership of the research lifecycle from ideating novel algorithms to scaling large training runs, ensuring our AI evolves alongside the user to become a true extension of their will.

What You’ll Do

• Design, train, and evaluate novel foundation models optimized for reasoning, multimodal understanding, and extreme personalization.
• Pioneer research in continual learning, enabling models to adapt in real-time based on local user interaction without catastrophic forgetting.
• Develop advanced techniques for data efficiency, alignment (e.g., RLHF), and parameter-efficient fine-tuning (PEFT) targeting our bespoke personal hardware.
• Partner directly with the infrastructure team to rapidly scale and unblock experimental architectures across large compute clusters.

Skills & Qualifications

Minimum Qualifications:

• BS, MS, or Ph.D. in Computer Science, Machine Learning, Mathematics, or equivalent practical industry experience.
• Deep understanding of modern deep learning architectures (e.g., Transformers, diffusion models) and advanced training methodologies.
• Extensive hands-on experience with PyTorch or JAX, with a track record of writing clean, scalable code for model training.
• Proven ability to take open-ended research problems from mathematical formulation to working, scaled implementations.
• A highly collaborative mindset and a bias for action to push boundaries in a fast-paced environment.

Preferred Qualifications: (We encourage you to apply even if you don't meet all of these)

• Track record of impactful publications at top-tier AI conferences (e.g., NeurIPS, ICLR, ICML) or equivalent industry research breakthroughs.
• Specific expertise in continual learning, reinforcement learning, or optimizing models for edge/on-device inference.
• Experience pre-training or aligning frontier-scale language or multimodal models.
• Compensation: Depending on experience and skills the expected base pay is $200,000 - $420,000 USD per year, plus equity.
• Benefits: Comprehensive health, dental, and vision insurance; unlimited PTO; and relocation assistance as needed.
• Visa Sponsorship: We sponsor visas and are committed to supporting the process for the right candidate.