Foundation Models Research Engineer - Personal AI

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

Benefits offered by this job

Health, dental, and vision insurance

Unlimited PTO

Relocation assistance

Visa sponsorship

Job summary

River AI in Palo Alto seeks exceptional researchers to design and train foundation models powering personal AI that runs on local hardware. You will own the research lifecycle from ideation to scaling large training runs, ensuring AI evolves with the user to become an extension of their will.

You’ll push the frontier of deep learning, focusing on architectures capable of continual learning, personalization, and efficient local inference, collaborating with infrastructure to scale experiments on

Qualifications

• BS, MS, or Ph.D. in Computer Science, Machine Learning, Mathematics, or equivalent practical industry experience.
• Deep understanding of modern deep learning architectures (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.

Responsibilities

• 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 bespoke personal hardware.
• Partner directly with the infrastructure team to rapidly scale and unblock experimental architectures across large compute clusters.

Skills

Deep learning architectures

Continual learning

RLHF

PyTorch/JAX

Collaborative mindset

Education

BS/MS/PhD in CS/ML/Math

Tools

PyTorch

JAX

Job description

River AI in Palo Alto seeks exceptional researchers to design and train foundation models powering personal AI that runs on local hardware. You will own the research lifecycle from ideation to scaling large training runs, ensuring AI evolves with the user to become an extension of their will.

You’ll push the frontier of deep learning, focusing on architectures capable of continual learning, personalization, and efficient local inference, collaborating with infrastructure to scale experiments on