Senior Research Scientist

Sciforium; San Francisco (CA); On-site; USD 130,000 - 160,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

An AI infrastructure company is seeking a Senior Research Scientist specializing in advanced AI/ML to lead cutting-edge research in large language models and generative media. The ideal candidate will have a PhD and over 5 years of relevant experience, with strong coding skills in Python and expertise in frameworks like PyTorch and JAX. This role offers competitive salary, equity, and comprehensive benefits including healthcare and flexible time off.

Qualifications

• PhD is required.
• 5+ years of research experience or exceptional fresh PhDs accepted.
• Proven track record of impactful research.

Responsibilities

• Lead research in advanced machine learning areas including LLMs and generative AI.
• Design and evaluate ML algorithms using frameworks like PyTorch and JAX.
• Conduct large-scale distributed training experiments.

Skills

Python

PyTorch

JAX

Machine Learning

Deep Learning

Distributed Training

Education

PhD in Computer Science, Machine Learning, AI, Mathematics, or a related field

Tools

Flax

XLA

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.
• 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.