Job Closed
This listing is no longer active.
Staff Python/PyTorch Developer – Frontend Inference Compiler
Location
United Arab Emirates
Posted
123 days ago
Salary
0
Seniority
Lead
Job Description
Staff Python/PyTorch Developer – Frontend Inference Compiler
Cerebras
• Analysis of new models from generative AI field and understanding of impacts on compilation stack • Develop and maintain model definition framework that consists of model building blocks to represent large language models based on PyTorch and Cerebras dialects ready to be deployed on Cerebras hardware. • Develop and maintain the frontend compiler infrastructure that ingests PyTorch models and produces an intermediate representation (IR). • Extend and optimize PyTorch FX / TorchScript / TorchDynamo-based tooling for graph capture, transformation, and analysis. • Collaboration with other teams throughout feature implementation • Research on new methods for model optimization to improve Cerebras inference
Job Requirements
- Degree in Engineering, Computer Science, or equivalent in experience and evidence of exceptional ability
- Strong Python programming skills and in-depth experience with PyTorch internals (e.g., TorchScript, FX, or Dynamo).
- Solid understanding of computational graphs, tensor operations, and model tracing.
- Experience building or extending compilers, interpreters, or ML graph optimization frameworks.
- Experience working with PyTorch and HuggingFace Transformers library
- Knowledge and experience working with Large Language Models (understanding Transformer architecture variations, generation cycle, etc.)
- Strong C++ programming skills.
- Knowledge of MLIR based compilation stack
Benefits
- Build a breakthrough AI platform beyond the constraints of the GPU.
- Publish and open source their cutting-edge AI research.
- Work on one of the fastest AI supercomputers in the world.
- Enjoy job stability with startup vitality.
- Our simple, non-corporate work culture that respects individual beliefs.
Related Guides
Related Job Pages
More Backend Engineer Jobs
Senior Cloud/Backend Engineer
Triple Whale 🐳All your data in one easy-to-use dashboard. We’re building the operating system for eCom.
• Deploy and support our service infrastructure in Kubernetes • Identify the right tools and technologies for major initiatives and then build them • Help other teams and developers design robust and scalable systems. • Scale and optimize multiple databases • Build internal tooling that accelerates developer velocity • Provide observability, monitoring, and visibility across our systems
• Designing and developing full stack applications using .NET technologies, with MVC-based front-end architecture • Modernizing and migrating legacy Python-based logic into scalable .NET solutions • Building automation tools and improving operational workflows • Writing clean, maintainable, and well-tested code with a strong emphasis on unit testing (targeting high code coverage) • Collaborating closely with architecture leadership to align with system design, standards, and long-term technical direction • Contributing to continuous improvement of development processes, tooling, and application performance • Supporting and maintaining existing applications, including occasional work with legacy automation tools
Senior Back-end Software Engineer, Java, willing to learn Go
Software MindSoftware House focused on results since 1999
• Focus on completing the remaining features and improving system performance • Ensure the platform meets production standards • Contribute to the ongoing product development, enhancements, and maintenance • Support the platform’s scalability and reliability
Senior Software Engineer, Backend
Customer.ioEmail, push notifications, text messages, in-app messages, webhooks: automated and powered by your data.
• Design and build distributed systems in Go that help customers send timely, relevant messages at scale • Collaborate across teams to debug issues and deliver reliable solutions • Own problems end to end - from architecture and schema design to testing, deployment, and monitoring • Share knowledge and raise the bar through sharing your progress publicly with short videos, thoughtful writing, and mentorship • Leverage AI tools to prototype, move faster, and make better decisions



