Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

ML Software Tool Development Engineer [Remote]

Full-time

cerebras

Sunnyvale, CA
  • Remote job

Cerebras Systems builds the world's largest AI chip, 56 times larger than GPUs. This architecture allows Cerebras to deliver industry-leading training and inference speeds; over 10 times faster than GPU-based hyperscale cloud inference services.

This order of magnitude increase in speed is transforming the user experience of AI applications, unlocking real-time iteration and increasing intelligence via additional agentic computation.

Cerebras works with the leading model labs, global enterprises, and cutting-edge AI-native startups. OpenAI recently announced a multi-year partnership with Cerebras, to deploy 750 megawatts of scale, transforming key workloads with ultra high-speed inference.

Responsibilities:

  • Lead the design and implementation of system-level debugging, validation, and observability platforms.

  • Develop automated systems for collecting and analyzing numerical, and execution anomalies.

  • Create visualization and analysis tools to enable efficient root-cause investigation.

  • Build frameworks for failure classification, regression detection, and anomaly monitoring.

  • Extend compilers, runtimes, and programming interfaces to support advanced profiling and instrumentation.

  • Improve system bring-up, low-level debug, and validation workflows.

  • Partner cross-functionally with compiler, hardware, firmware, runtime, and infrastructure teams.

  • Establish best practices for debuggability, reliability, and operational excellence.

  • Lead high-impact initiatives.

  • Support incident response and drive long-term corrective actions.

Skills & Qualifications

  • Strong proficiency in C++ and Python, with a track record of building reliable, high-performance systems and tooling.

  • Demonstrated experience debugging complex hardware/software systems and driving issues to root cause.

  • Experience analyzing system-level data structures, execution graphs, or dependency networks for diagnostics and validation.

  • Proven ability to design and build intuitive visualization and analysis tools for complex technical data.

  • Experience with compiler internals, custom hardware interfaces, or low-level protocol design.

  • Strong written and verbal communication skills, with the ability to explain technical concepts to diverse stakeholders.

  • Ability to work independently and lead complex technical projects end-to-end.

Preferred Skills & Qualifications

  • Familiarity with machine learning training and inference pipelines, especially distributed training and large-model scaling.

  • Prior work on high-performance clusters, HPC systems, or custom hardware/software co-design.

 

Why Join Cerebras

People who are serious about software make their own hardware. At Cerebras, we have built a breakthrough architecture that is unlocking new opportunities for the AI industry. With dozens of model releases and rapid growth, we’ve reached an inflection point in our business. Members of our team tell us there are five main reasons they joined Cerebras:

  1. Build a breakthrough AI platform beyond the constraints of the GPU.

  2. Publish and open source their cutting-edge AI research.

  3. Work on one of the fastest AI supercomputers in the world.

  4. Enjoy job stability with startup vitality.

  5. Our simple, non-corporate work culture that respects individual beliefs.

Find out more about what it's like to work at Cerebras here !

Apply today and become part of the forefront of groundbreaking advancements in AI!

Cerebras Systems is committed to creating an equal and diverse environment and is proud to be an equal opportunity employer. We celebrate different backgrounds, perspectives, and skills. We believe inclusive teams build better products and companies. We try every day to build a work environment that empowers people to do their best work through continuous learning, growth and support of those around them.

This website or its third-party tools process personal data. For more details, click here to review our CCPA disclosure notice.

Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the ML Software Tool Development Engineer [Remote] in Sunnyvale, CA vacancy
  • $144.6k - $263.8k

     ...decisionsCraft transformative tools to drive design intelligence...  ...of doing things.Support development of automated ECAD and DRC checks...  ...in mechanical, computer or software engineering (or similar)Preferred...  ...and web APIsUnderstanding of ML and deep learning or reinforcement... 
    Suggested
    Relocation
    Flexible hours

    Apple

    Cupertino, CA
    3 hours ago
  •  ...together, we’ll advance your career.THE ROLEAMD is looking for a Software Development Engineer to help build the next generation of integrated marketing...  ..., or data-driven products.Practical experience with AI/ML, automation, or AI-assisted software development.Strong analytical... 
    Suggested

    AMD

    Santa Clara, CA
    4 days ago
  • $148.4k

     ...design, firmware, compiler, runtime, and ML engineers to develop software that enables and optimizes custom ML...  ..., runtime libraries, profiling tools, simulation frameworks, or chip validation...  ...-level and performance-critical development Develop CI/CD pipelines, test... 
    Suggested
    Internship
    Local area
    Worldwide

    Amazon Jobs

    Cupertino, CA
    3 days ago
  • $188.5k - $265k

     ...Principal Software Development EngineerZscaler accelerates digital transformation so customers can...  ...role, reporting to the Director, Software Engineering in the Engineering department. You will...  ...understanding of AI/ML technologies and experience leveraging,... 
    Suggested
    Full time
    Work at office
    Local area
    Worldwide

    Zscaler

    Santa Clara, CA
    2 days ago
  • $164k - $295k

     ...CrowdStrike is seeking an experienced Software Development Engineer to join our AI Detection and Response...  ...monitoring, profiling and optimization tools Bonus Points: Previous experience...  ...products Deep understanding of AI/ML security challenges, threats and mitigation... 
    Suggested
    Permanent employment
    Work experience placement
    Work at office
    Local area
    Worldwide
    2 days per week
    3 days per week
    Sunnyvale, CA
    a month ago
  • $21 - $40 per hour

     ...HQ in Northwest Arkansas! Software Engineer Intern: We're looking for...  ...skills to create innovative tools and system enhancements to drive...  ...Boot, Cosmos DB, Docker, AI/ML and many other technologies...  ...in the full software development lifecycle, including the discovery... 
    Hourly pay
    Minimum wage
    Summer work
    Internship
    Summer internship
    Work at office
    Relocation
    Home office

    Walmart

    Sunnyvale, CA
    3 days ago
  • $70 per hour

     ...in simulation across 15+ U.S. states. Software Engineering builds the brains of Waymo's fully autonomous...  ...will be: The Model Optimization & ML Runtime team (within Smart Perception /...  ...software engineering and deep learning development skills in Python and modern frameworks... 
    Hourly pay
    Full time
    Internship
    Summer internship
    Relocation package

    Waymo

    Mountain View, CA
    4 days ago
  • $109.65k

     ...At Amazon, our tech isn't just a tool, it's your playground. Our engineers work on scalable systems, cloud services...  ...ambiguity in an environment where development cycles are measured in weeks, not...  ...control systems Knowledge of software development lifecycle, including... 
    Full time
    Summer work
    Internship
    Seasonal work
    Summer internship
    Work at office
    Local area
    Worldwide
    Monday to Friday

    Amazon

    Sunnyvale, CA
    4 days ago
  • $70 per hour

     ...in simulation across 15+ U.S. states. Software Engineering builds the brains of Waymo's fully autonomous...  ...and Machine Learning (SIML) Subteam: ML Frameworks and Efficiency...  ...PyTorch, or TensorFlow) Strong software development skills with Python We prefer: Has... 
    Hourly pay
    Full time
    Internship
    Summer internship
    Relocation package

    Waymo

    Mountain View, CA
    1 day ago
  • $70 per hour

     ...and tens of billions in simulation across 15+ U.S. states. Software Engineering builds the brains of Waymo's fully autonomous driving technology...  ..., or real-time systems Experience with large-scale data/ML applications Note: This will be a hybrid onsite... 
    Hourly pay
    Full time
    Internship
    Summer internship

    Waymo

    Mountain View, CA
    1 day ago
  • $185.5k - $265k

     ...Role We are looking for a Principal Software Development Engineer (Microservices) to join our team. This...  ...) ~ Foundational understanding of AI/ML technologies and experience leveraging...  ...strategies, and data replay tooling OSS contributions, technical talks,... 
    Full time
    Contract work
    Work at office
    Local area
    3 days per week

    Zscaler

    Santa Clara, CA
    5 days ago
  • $142k

     ...individuals to join its dynamic design engineering team as an AI-Driven Physical...  ...Research and implement AI/ML techniques to identify new...  ...Proficiency in industry-standard tools for physical design, including...  ...route, and timing analysis software. Familiarity with scripting... 
    Internship
    Local area
    Immediate start
    Shift work

    Intel

    Santa Clara, CA
    11 hours ago
  • $150.4k - $277.6k

    Role Number: 200646833Team: Software and ServicesLocation: CupertinoWeekly Hours: 40Posted...  ...looking for an outstanding Software Development Engineer to join our team and revolutionize how...  ...culture, our benefits and our digital tools. By welcoming as many perspectives as possible... 
    Relocation

    Apple

    Cupertino, CA
    3 hours ago
  • $193.3k - $261.5k

     ...for a Senior SoC Modeling Engineer to join the team and...  ...models, infrastructure, and tooling for our customers.As part of the ML accelerator modeling...  ...verification, emulation, and software teams to build, debug,...  ...used in both our SoC development process and by our partner... 
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    2 days ago
  •  ...technologies and advanced engineering principles to drive...  ...goals, scope, and own development efforts while...  ...integrating graph compilers. Software Engineering Best Practices...  ..., leveraging tools like Compute Kernel (CK...  ...accelerated compute into ML frameworks (e.g., PyTorch... 

    AMD

    Santa Clara, CA
    2 days ago
  • $165.2k - $223.6k

    We are seeking an experienced engineer to work on distributed AI/ML systems. This role involves working on collective...  ...of AWS and develops hardware and software components that are critical...  ...non-internship professional software development experience- 2+ years of non-... 
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    3 days ago
  • $165.2k - $223.6k

     ...access to AI hardware and software infrastructure. In...  ...performance of complex ML models executed on AWS...  ...role is for a software engineer in the Compiler team for...  ...developers, implement tools to analyze numerical errors...  ...work in a startup-like development environment, where you’... 
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    2 days ago
  • $193.3k - $261.5k

     ...customized stack of hardware, firmware, and software to deliver unparalleled virtualization...  ...looking for an experienced software engineer to drive development for new EC2 machine learning platforms...  ...modern cloud computing and AI/ML workloads.Key job responsibilitiesThe... 
    Internship
    Local area
    Flexible hours

    Amazon

    Santa Clara, CA
    4 days ago
  • $193.3k - $261.5k

     ...The EC2 Nitro local storage team is seeking an experienced Software Development Engineer to build virtualized, hardware-accelerated solutions for EC...  ...and data warehousing systems that power state-of-the-art AI/ML infrastructure.This role offers unique opportunities to... 
    Internship
    Local area
    Flexible hours

    Amazon

    Santa Clara, CA
    11 hours ago
  • $148.4k

     ...learning servers, and we are hiring a software engineer to help build the Hardware Abstraction...  ...background is required (you will pick up ML concepts from colleagues if you want to...  ...non-internship) in professional software development, or currently enrolled in or recently... 
    Internship
    Local area

    Amazon

    Cupertino, CA
    2 days ago
  • $165.2k - $223.6k

     ...of hardware, firmware, and software to deliver unparalleled virtualization...  ...an experienced software engineer to drive development for new EC2 machine...  ...required. In-depth knowledge of ML frameworks and cluster...  ...distributed software applications, tools, systems, and services... 
    Internship
    Local area
    Flexible hours

    Amazon

    Santa Clara, CA
    4 days ago
  • $150k - $183k

    Job-ID: 24397Category: Software DevelopmentLocation: Sunnyvale, CA, United...  ...are hiring a Machine Learning Engineer to own the detection models...  ...prompts, responses, and tool-call payloads — dataset construction...  ...experience in adversarial ML or LLM security.Must be authorized... 
    Full time
    Worldwide

    Fortinet

    Sunnyvale, CA
    3 days ago
  • $127.1k - $185k

     ...builds the AWS Neuron SDK, the software development kit used to accelerate deep...  ...for a wide range of ML models and model architectures...  ...hardware-software boundary, our engineers build systematic infrastructure...  ..., and help deploy agentic tools to accelerate the enablement... 
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    1 day ago
  • $129.3k - $194.7k

    Role Number: 200693291Team: Software and ServicesLocation: CupertinoWeekly...  ...Apple Ads platform. As a Tooling Engineer, you'll help develop tooling...  ...applying AI/ML concepts to build robust tools...  ...with Git and basic software development workflowsStrong problem-solving... 
    Relocation

    Apple

    Cupertino, CA
    3 hours ago
  • $165.2k - $223.6k

    AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale...  ...servers that use them.As the Software Development Engineer for the Neuron Runtime Team, you will...  ...system behavior. Improving performance of ML Kernels and ML Frameworks.In this role,... 
    Internship
    Local area
    Work from home
    Flexible hours

    Amazon

    Cupertino, CA
    3 days ago
  •  ...ll join a diverse team of software, hardware, and network engineers, supply chain specialists...  ...a wide gamut of tools to help planning/scaling/...  ...would like to talk to you.At ML-Navigator team we are responsible...  ...engineers. As a software development engineer, you will:Help... 
    Internship
    Local area
    Worldwide
    Shift work

    Amazon

    Santa Clara, CA
    2 days ago
  • $193.3k - $261.5k

     ...improvements with minimal engineering effort from partner...  ...infrastructure for AI/ML workloads through centralized scheduling, software performance profiling,...  ...across Amazon.As a Software Development Engineer in Alloy, you...  ...and profiling tools.Key job responsibilities... 
    Internship
    Local area
    Flexible hours

    Amazon

    Sunnyvale, CA
    4 days ago
  • $70 per hour

     ...simulation across 15+ U.S. states. Software Engineering builds the brains of Waymo's...  ...data exploration tools and visualization dashboards...  ...remaining Strong software development experience in Python and/or...  ...published papers in top-tier AI/ML, data mining, or computer... 
    Hourly pay
    Full time
    Internship
    Summer internship
    Relocation package

    Waymo

    Mountain View, CA
    3 days ago
  • $164k - $295k

     ...CrowdStrike is seeking an experienced Engineering Manager to lead a cloud...  ...reviews and support career development Remove blockers, shield...  ...You'll Need: ~10+ years of software development experience building...  ...Deep understanding of AI/ML security challenges, threats,... 
    Permanent employment
    Work experience placement
    Work at office
    Local area
    Worldwide
    2 days per week
    3 days per week
    Sunnyvale, CA
    a month ago
  • $169k - $218k

     ...for a foundational full-stack engineer to help drive Gridmatic’s initiative...  ...business uses long-term ML forecasting to sign energy supply...  ...usage Building internal tools to allow us to do complex energy...  ...if you: Have 3+ years of software engineering experience Have... 
    Contract work
    Work experience placement

    Gridmatic

    Cupertino, CA
    2 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to ML Software Tool Development Engineer [Remote]. Be the first to apply!