Machine Learning Engineer, Staff

7 days ago


Santa Clara, United States d-Matrix Full time

d-Matrix has fundamentally changed the physics of memory-compute integration with our digital in-memory compute (DIMC) engine. The “holy grail” of AI compute has been to break through the memory wall to minimize data movements. We’ve achieved this with a first-of-its-kind DIMC engine. Having secured over $154M, $110M in our Series B offering, d-Matrix is poised to advance Large Language Models to scale Generative inference acceleration with our chiplets and In-Memory compute approach. We are on track to deliver our first commercial product in 2024. We are poised to meet the energy and performance demands of these Large Language Models.

The company has 100+ employees across Silicon Valley, Sydney and Bengaluru.

Our pedigree comes from companies like Microsoft, Broadcom, Inphi, Intel, Texas Instruments, Lucent, MIPS and Wave Computing. Our past successes include building chips for all the cloud hyperscalers globally - Amazon, Facebook, Google, Microsoft, Alibaba, Tencent along with enterprise and mobile operators like China Mobile, Cisco, Nokia, Ciena, Reliance Jio, Verizon, AT&T. We are recognized leaders in the mixed signal, DSP connectivity space, now applying our skills to next generation AI.

Location: Hybrid, working onsite at our Santa Clara, CA headquarters 3 days per week.

Position: d-Matrix is seeking a Machine Learning Engineer to join our Algorithm Team. We’re looking for someone to invent, design, and implement efficient algorithms that will be used to optimize Large Language Model inference on DNN Accelerators we develop. You would be part of a close-knit team of mathematicians, ML researchers, and ML engineers who create and apply advanced algorithmic and numerical techniques to the most cutting-edge and high-impact research in the overlap of mathematics, ML, and modern LLM applications.

What you will do:

  • Implement advanced quantization algorithms for modern LLMs.
  • Support and develop numerical libraries.
  • Participate in design of the next generation chips for multi-model training and inference.

What you will bring:

  • MSc or PhD in CS, statistics, physics, or a related STEM field.
  • 4+ years of experience in industrial coding and OOP design.
  • High motivation to innovate and collaborate with experts from various fields.
  • Experience in Python.
  • Experience with transformer architecture is advantageous but not mandatory.

Equal Opportunity Employment Policy

d-Matrix is proud to be an equal opportunity workplace and affirmative action employer. We’re committed to fostering an inclusive environment where everyone feels welcomed and empowered to do their best work. We hire the best talent for our teams, regardless of race, religion, color, age, disability, sex, gender identity, sexual orientation, ancestry, genetic information, marital status, national origin, political affiliation, or veteran status. Our focus is on hiring teammates with humble expertise, kindness, dedication and a willingness to embrace challenges and learn together every day.

d-Matrix does not accept resumes or candidate submissions from external agencies. We appreciate the interest and effort of recruitment firms, but we kindly request that individuals interested in opportunities with d-Matrix apply directly through our official channels. This approach allows us to streamline our hiring processes and maintain a consistent and fair evaluation of all applicants. Thank you for your understanding and cooperation.

#J-18808-Ljbffr

  • Santa Clara, California, United States Amazon Full time

    About the RoleWe are seeking a skilled Machine Learning Engineer to join our team at Amazon. As a Machine Learning Engineer, you will be responsible for designing, developing, and deploying machine learning models to solve complex business problems. You will work closely with data scientists, software engineers, and other stakeholders to identify...


  • Santa Clara, California, United States XPENG Motors Full time

    We are seeking a highly skilled Staff Machine Learning Engineer - AI Foundation to join our team at XPeng Motors.The ideal candidate will have a deep understanding of large-scale deep learning models and experience with PyTorch.Responsibilities include designing, training, and deploying large deep learning models that can leverage vast amounts of labeled and...


  • Santa Clara, United States Acceler8 Full time

    Machine Learning EngineerWe are seeking a Machine Learning Engineer to join a very well-funded (Series D) autonomous vehicle startup based in Mountain View, CA. Our mission is to reimagine the idea that people have of the autonomous vehicle industry today by creating a positive real-world impact.Within this role, you will be working alongside our highly...


  • Santa Clara, United States Eightfold LLC Full time

    About Eightfold.ai Eightfold.ai is at the forefront of innovation in the AI-driven HR tech space. We're not just building software; we're shaping the future of how organizations find, manage, and empower their talent. Our groundbreaking AI platform is revolutionizing the industry, and we're looking for exceptional engineers to join our team and drive the...


  • Santa Clara, California, United States Amazon Development Center U.S., Inc. Full time

    About the RoleWe are seeking a highly skilled Senior Machine Learning Engineer to join our team at Amazon Development Center U.S., Inc. The ideal candidate will have a strong background in machine learning, large language models, and multimodal models, with experience in programming languages such as Java, C++, and Python.As a Senior Machine Learning...


  • Santa Clara, United States Skale Full time

    We're hiring for a Series B furniture AI start-up in the Bay Area. They're looking for a Senior Machine Learning Engineer to drive innovation in the furniture tech world, working on a variety of computer vision and machine learning problems.You'll get a chance to take ownership and work on the full machine learning life cycle, developing and deploying...


  • santa clara, United States Skale Full time

    We're hiring for a Series B furniture AI start-up in the Bay Area. They're looking for a Senior Machine Learning Engineer to drive innovation in the furniture tech world, working on a variety of computer vision and machine learning problems.You'll get a chance to take ownership and work on the full machine learning life cycle, developing and deploying...


  • Santa Clara, California, United States Palo Alto Networks Full time

    Protecting Our Digital FuturePalo Alto Networks is the fastest-growing security company in history, and we're looking for a talented Machine Learning Software Engineer to join our team. As a key member of our Data Science team, you'll use data science and machine learning to solve complex problems and develop innovative solutions to address cyber security...


  • Santa Clara, United States Servicenow Full time

    Company DescriptionIt all started in sunny San Diego, California in 2004 when a visionary engineer, Fred Luddy, saw the potential to transform how we work. Fast forward to today — ServiceNow stands as a global market leader, bringing innovative AI-enhanced technology to over 8,100 customers, including 85% of the Fortune 500. Our intelligent cloud-based...


  • Santa Clara, United States TCWGlobal Full time

    Sr. Machine Learning EngineerW2 Contract 6 MonthsRemote, USA$90/hr + Benefits, PTO, Paid Holidays As a Data / ML Engineer, you'll work alongside our Digital Marketing platform team, dedicated to addressing our evolving data-driven marketing and compliance challenges. We value expertise in data science paired with a robust data engineering foundation....


  • santa clara, United States TCWGlobal Full time

    Sr. Machine Learning EngineerW2 Contract 6 MonthsRemote, USA$90/hr + Benefits, PTO, Paid Holidays As a Data / ML Engineer, you'll work alongside our Digital Marketing platform team, dedicated to addressing our evolving data-driven marketing and compliance challenges. We value expertise in data science paired with a robust data engineering foundation....


  • santa clara, United States Meshy Full time

    About MeshyHeadquartered in the Silicon Valley, Meshy is a leading 3D generative AI company on a mission to Unleash 3D Creativity. We simplify the creation of distinctive 3D assets for both professional artists and hobbyists by transforming text and images into stunning 3D models in minutes.Our global team of 30 experts in computer graphics, AI, and art...


  • Santa Clara, United States Meshy Full time

    About MeshyHeadquartered in the Silicon Valley, Meshy is a leading 3D generative AI company on a mission to Unleash 3D Creativity. We simplify the creation of distinctive 3D assets for both professional artists and hobbyists by transforming text and images into stunning 3D models in minutes.Our global team of 30 experts in computer graphics, AI, and art...


  • santa clara, United States Meshy Full time

    About MeshyHeadquartered in the Silicon Valley, Meshy is a leading 3D generative AI company on a mission to Unleash 3D Creativity. We simplify the creation of distinctive 3D assets for both professional artists and hobbyists by transforming text and images into stunning 3D models in minutes.Our global team of 30 experts in computer graphics, AI, and art...


  • Santa Clara, California, United States Palo Alto Networks Full time

    About the RolePalo Alto Networks is seeking a highly skilled Principal Machine Learning Engineer to join our team. As a key member of our cybersecurity team, you will be responsible for designing and developing advanced machine learning solutions to protect our customers' networks.Key Responsibilities:Design and develop machine learning models to detect and...


  • Santa Clara, California, United States Eightfold LLC Full time

    About Eightfold Eightfold AI is the industry leader in AI-powered talent intelligence and transforming the way organizations manage their talent. Our AI-powered Talent Intelligence Platform helps companies identify, attract, and retain top talent, while also providing employees with the tools they need to grow and succeed in their careers. About the AI/ML...


  • Santa Clara, California, United States Amazon Full time

    Amazon - An Industry Leader in E-commerce and Technology We are seeking talented Senior Machine Learning Engineers to join our team at Amazon, working on various AI projects that enhance the shopping experience for our customers. As a Senior Machine Learning Engineer, you will be responsible for designing, developing, and deploying large-scale machine...


  • Santa Clara, California, United States Palo Alto Networks Full time

    About the RoleWe are seeking a highly skilled Principal Machine Learning Engineer to join our team at Palo Alto Networks. As a key member of our cybersecurity team, you will be responsible for designing and developing advanced AI and machine learning solutions to protect our customers' digital way of life.Your ResponsibilitiesDesign and develop workflow...


  • Santa Clara, United States Eightfold LLC Full time

    Job Location: Santa Clara, CA (This is NOT a remote position ) About Eightfold Eightfold AI is the industry leader in AI-powered talent intelligence and transforming the way organizations manage their talent. Our AI-powered Talent Intelligence Platform™ helps companies identify, attract, and retain top talent, while also providing employees with the...


  • Santa Clara, California, United States Precisionneuro Full time

    About Precision NeurosciencePrecision Neuroscience is a pioneering company that is revolutionizing the field of brain-computer interfaces (BCIs). Our mission is to develop cutting-edge technologies that enable people with neurological conditions to regain independence and communicate with loved ones.We are seeking an experienced Director of Machine Learning...