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

Bachelor DegreeEnglishPythonPyTorch

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 Job Pages

More Backend Engineer Jobs

Triple Whale 🐳 logo

Senior Cloud/Backend Engineer

Triple Whale 🐳

All your data in one easy-to-use dashboard. We’re building the operating system for eCom.

Backend Engineer123 days ago
Full TimeRemoteTeam 51-200H1B No Sponsor

• 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

United States
$147K - $187K / year
Job Closed
Software Mind logo

Senior Full-Stack Engineer – .NET

Software Mind

Software House focused on results since 1999

Backend Engineer123 days ago
Full TimeRemoteTeam 1,001-5,000Since 1999H1B No Sponsor

• 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

Costa Rica
Job Closed
Full TimeRemoteTeam 1,001-5,000Since 1999H1B No Sponsor

• 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

Romania
Job Closed
Customer.io logo

Senior Software Engineer, Backend

Customer.io

Email, push notifications, text messages, in-app messages, webhooks: automated and powered by your data.

Backend Engineer123 days ago
Full TimeRemoteTeam 51-200Since 2012H1B No Sponsor

• 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

Europe
$150K - $200K / year