Senior Staff Engineer- AIML Software & Compilers

About Marvell

Marvells semiconductor solutions are the essential building blocks of the data infrastructure that connects our world. Across enterprise cloud and AI automotive and carrier architectures our innovative technology is enabling new possibilities.

At Marvell you can affect the arc of individual lives lift the trajectory of entire industries and fuel the transformative potential of tomorrow. For those looking to make their mark on purposeful and enduring innovation above and beyond fleeting trends Marvell is a place to thrive learn and lead.

Your Team Your Impact

What You Can Expect

• As a Sr Staff Engineer Machine Learning Software/Firmware Engineering you willinfluence and contribute to the architecture design and implementation of AI infrastructure and ML compiler for the Marvell Machine learning acceleration and Infrastructure. Work closely with Machine learning networking and storage experts to develop compiler and runtime infrastructure to achieve the targeted features and performance benchmarks.

• With your in-depth technical knowledge experience and understanding you will efficiently apply solutions to resolve complex technical problems. Creating state-of-the-art data center infrastructure solutions and IPs that make a significant impact on Marvells Accelerated Compute mission.

• Architect design and develop system software for reliable scalable high performing AI/ML systems using Python C/C.

• Design and develop the compiler runtime and frameworks for Marvell ML Infrastructure and acceleration solutions.

• Evaluate code performance debug diagnose and drive resolution using advanced state of art technologies.

• Work on developing and enhancing the AI acceleration plugins for heterogeneous systems.

• Communicate and collaborate effectively with cross-functional hardware and software engineering teams.

• Develop software and troubleshoot issues related to hardware bring-up multi-threaded software development with virtualized hardware.

• Hands on experience on hyperscale workloads such as Ranking and Recommendation Systems Large Language Models Retrieval-Augmented Generation etc accelerated by a broad range of custom compute connectivity storage security technologies

• Develop and contribute to open frameworks such as LLVM MLIR OpenCL or OpenXLA.

• Plan schedule estimate efforts for projects and provide technical guidance to junior engineers during execution.

What Were Looking For

• BS/MS degree in: Computer Science Computer Engineering or Electrical Engineering
• Minimum 8-10 years of experience in architecting and developing compilers and system software.
• Must possess strong hands-on experience in C design and development principles.
• Experience designing and developing code on Python especially Python 3.x
• Understanding and working knowledge of CXL memory systems in Heterogeneous computing is a big plus.
• Hands on experience on hyperscale workloads such as Ranking and Recommendation Systems Large Language Models Retrieval-Augmented Generation etc. accelerated by a broad range of custom compute connectivity storage security technologies
• Experience working with TensorFlow PyTorch and JAX would be a big plus.
• Experience working on and contributing to an active compiler toolchain codebase such as LLVM MLIR OpenCL or OpenXLA frameworks.
• Experience and familiarity with ML accelerators GPU and ARM architectures is a plus.
• Experience with setting up AI/ML workloads and benchmarking MLPerf or similar will be an added advantage.
• Comfortable in GIT Jira and Confluence.
• Must have the ability to multi-task in a fast-paced environment where the workloads evolve at a rapid pace.

Additional Compensation and Benefit Elements

With competitive compensation and great benefits you will enjoy our workstyle within an environment of shared collaboration transparency and inclusivity. Were dedicated to giving our people the tools and resources they need to succeed in doing work that matters and to grow and develop with us. For additional information on what its like to work at Marvell visit our Careers page.

All qualified applicants will receive consideration for employment without regard to race color religion sex national origin sexual orientation gender identity disability or protected veteran status.

Interview Integrity

As part of our commitment to fair and authentic hiring practices we ask that candidates do not use AI tools (e.g. transcription apps real-time answer generators like ChatGPT CoPilot or note-taking bots) during interviews.

Our interviews are designed to assess your personal experience thought process and communication skills in real-time. If a candidate uses such tools during an interview they will be disqualified from the hiring process.

This position may require access to technology and/or software subject to U.S. export control laws and regulations including the Export Administration Regulations (EAR). As such applicants must be eligible to access export-controlled information as defined under applicable law. Marvell may be required to obtain export licensing approval from the U.S. Department of Commerce and/or the U.S. Department of State. Except for U.S. citizens lawful permanent residents or protected individuals as defined by 8 U.S.C. 1324b(a)(3) all applicants may be subject to an export license review process prior to employment.