Sciforium
Model Implementation Engineer
San Francisco
Sponsorship not specifiedDetected 78 days ago
PythonFull-Stack DevelopmentMachine LearningDeep LearningPyTorchNLPLLMsElectrical EngineeringResearch
About the role
- We are seeking a highly skilled Model Implementation Engineer who is passionate about bringing cutting-edge machine learning models into production-ready systems.
- You will work at the intersection of research and systems, translating the latest ideas into robust, scalable implementations.
- This includes collaborating closely with GPU kernel and systems teams to ensure models are efficiently executed on modern accelerators.
Responsibilities
- Implement new model architectures and research ideas, ensuring correctness, scalability, and production readiness.
- Rapidly integrate newly released open-source models to enable day-0 support across the platform.
- Collaborate closely with GPU kernel and systems teams to optimize model execution and improve overall performance.
- Develop and maintain internal tooling, testing frameworks, and documentation to support model reliability and reproducibility.
- Proven experience maintaining and developing model libraries or reusable ML components.
Requirements
- Experience with model performance optimization and profiling.
- Experience with JAX-first workflows and advanced features (e.g., pjit, xmap, or custom transformations).
Nice to have
- Hands-on experience with JAX and/or PyTorch (JAX strongly preferred).
Compensation
- Competitive salary and equity
Benefits
- Medical, dental, and vision insurance
- Flexible time off
- Competitive salary and equity
- Maintain and evolve a large-scale library of modern machine learning models, including but not limited to LLMs, ASR, TTS, image and video models, and diffusion-based systems.
- 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, Applied Mathematics, or a related field.
- Solid understanding of deep learning architectures across multiple domains (e.g., NLP, vision, speech, generative models).
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.
Visa & Work Authorization
- 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.
Apply directly at Sciforium →Create a free account for alerts like thisView Sciforium immigration profile
This listing is sourced directly from Sciforium's careers page and normalized into a canonical job model.