Senior Research Scientist

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

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

• Medical, dental, and vision insurance
• 401k plan
• Daily lunch, snacks, and beverages
• Flexible time off
• Competitive salary and equity

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.