Senior Staff Engineer-AI/ML

Job Description About Marvell Marvell's 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 Marvell is a leading provider of innovative technologies, including ultra-fast read channels, high-performance processors, leading edge transceivers, highly efficient analog designs, and powerful cryptographic engines. These solutions address all segments of the hard disk drive (HDD) and solid-state drive (SSD) electronics markets, providing complete solutions including controllers, product firmware, and reference board designs. Many of the same technologies have been utilized in Marvell system solutions products, powering PCs, servers, cloud, and enterprise systems. 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 Marvell's 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 We're 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. We're 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 it's like to work at Marvell, visit our 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. 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. #LI-MN1