Model Implementation Engineer

About the Company

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, the team is scaling rapidly to build the full stack powering frontier AI models and real-time applications.

Responsibilities

• Maintain and evolve a large-scale library of modern machine learning models, including LLMs, ASR, TTS, image, video, and diffusion-based systems
• Implement new model architectures and research ideas for production readiness
• Rapidly integrate newly released open-source models to enable day-0 support
• Collaborate with GPU kernel and systems teams to optimize model execution
• Benchmark models to ensure they meet performance, latency, and efficiency standards
• Develop and maintain internal tooling, testing frameworks, and documentation

Requirements

• At least 3 years of industry or research experience in model implementation or applied machine learning
• Master of Science or higher in Computer Science, Machine Learning, Electrical Engineering, or related field
• Strong programming skills in Python
• Hands-on experience with JAX and/or PyTorch (JAX strongly preferred)
• Proven experience maintaining model libraries or reusable ML components
• Solid understanding of deep learning architectures (NLP, vision, speech, generative models)
• Experience implementing models from research papers

Preferred Qualifications

• Experience with model performance optimization and profiling
• Familiarity with low-level performance considerations on GPUs/TPUs
• Experience with large-scale model training or inference systems
• Contributions to open-source model repositories or ML frameworks
• Experience with JAX-first workflows (pjit, xmap, or custom transformations)

Benefits

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