Senior AI Research Scientist - LLMs & Scalable Training

Sciforium; San Francisco (CA); On-site; USD 180,000 - 280,000; Full time

Benefits offered by this job

Medical, dental, and vision insurance

401k plan

Daily lunch, snacks, and beverages

Flexible time off

Competitive salary and equity

Job summary

Sciforium, an AI infrastructure company, seeks a Senior Research Scientist to push breakthroughs in large language models, generative AI, and scalable training systems. You will lead cutting-edge research, design novel algorithms using PyTorch and JAX, and coordinate with engineering teams to bring prototypes into production.

You will mentor junior researchers, publish in top venues, and contribute to building a world-class AI research culture at Sciforium.

Qualifications

• PhD in Computer Science, Machine Learning, AI, Mathematics or related field.
• 5+ years of academic or industry research experience.
• Proven track record with publications in top-tier conferences (NeurIPS/ICML/ICLR/CVPR/ICCV/ECCV).
• Strong coding proficiency in Python with PyTorch and/or JAX.

Responsibilities

• Lead research in advanced ML areas including LLMs, generative AI, modeling, optimization, diffusion and Transformer architectures.
• Design, implement, and evaluate novel ML algorithms using PyTorch and JAX.
• Conduct large-scale distributed training experiments on multi-GPU/TPU systems.
• Drive performance improvements via debugging, profiling, and pipeline optimizations.
• Produce high-quality research outputs: papers, reports, patents, and reproducible code.
• Collaborate with engineering/product teams to translate research into production systems.
• Stay ahead of research developments and integrate techniques into Sciforium's AI roadmap.
• Mentor junior researchers and contribute to a world-class AI research culture.

Skills

Python

PyTorch

JAX

Distributed training

Research leadership

Education

PhD in Computer Science / ML / AI

Tools

JAX/Flax/XLA stack

PyTorch

Job description

Sciforium is an AI infrastructure company developing next-generation multimodal AI models and a proprietary, high-efficiency serving platform. Backed by multi-million-dollar funding and direct sponsorship from AMD with hands-on support from AMD engineers the team is scaling rapidly to build the full stack powering frontier AI models and real-time applications.

About the role

Sciforium is seeking an exceptional Senior Research Scientist specializing in advanced AI/ML to push the boundaries of scientific and engineering innovation. In this role, you will lead cutting-edge research initiatives in large language models, generative media, model architecture, optimization, and scalable training systems. You will work hands-on with modern ML frameworks, contribute original research, and collaborate with engineering teams to bring impactful models into production. This role is ideal for a highly motivated researcher who is passionate about driving foundational breakthroughs in AI.

What you'll do

• Lead research in advanced machine learning areas such as LLMs, generative AI, foundational modeling, optimization techniques, diffusion models, and novel Transformer architectures.

• Design, implement, and evaluate novel ML algorithms using frameworks like PyTorch and JAX.

• Conduct large-scale distributed training experiments using multi-GPU/TPU systems and modern compute infrastructure.

• Drive performance improvements through framework debugging, speed optimization, and training pipeline enhancements.

• Produce high-quality research output including papers, internal reports, patents, and reproducible code.

• Collaborate with engineering and product teams to translate research prototypes into scalable production systems.

• Stay ahead of the latest research developments and integrate state-of-the-art techniques into Sciforium’s AI roadmap.

• Mentor junior researchers and contribute to building a world-class AI research culture.

Ideal candidate profile

• PhD in Computer Science, Machine Learning, AI, Mathematics, or a related field (required).

• 5+ years of academic or industry research experience (with flexibility for exceptional fresh PhDs from top programs).

• Proven track record of impactful research, evidenced by publications in NeurIPS, ICML, ICLR, CVPR, ICCV, ECCV, or similar top-tier venues.

• Strong coding proficiency in Python, with deep experience in PyTorch and/or JAX.

• Experience with debugging, performance profiling, and speed optimization.

• Expertise in distributed training at scale.

Nice-to-have

• Extensive experience with JAX/Flax/XLA stack

• Experience with efficient model serving, inference optimization, model compression, or production ML systems.

• Hands-on research or engineering experience with diffusion models or generative modeling.

• Experience contributing to product development or collaborating with cross-functional engineering teams.

• Open-source contributions to major ML frameworks or libraries.

Benefits include

• Medical, dental, and vision insurance

• 401k plan

• Daily lunch, snacks, and beverages

• Flexible time off

• Competitive salary and equity

Equal opportunity

Sciforium is an equal opportunity employer. All applicants will be considered for employment without attention to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.