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.

Requirements

Must-haves

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.
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.
Daily lunch, snacks, and beverages
Flexible time off
Competitive salary and equity

Nice-to-haves

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.

Application prep

Use the role’s language and application details while tailoring your materials.

Keywords to mirror

aimlmachine learningpython

Platform: ashby
Effort: quick

Benefits

Benefits detected in the listing.

Dental

Vision

401k