Software Engineer - Kernels
MatX is on a mission to be the compute platform for AGI. We are developing vertically integrated full-stack solutions from silicon to systems including hardware and software to train and run the largest ML workloads for AGI. We primarily use the Rust programming language .
Responsibilities include:
- Design and optimize kernels that interface directly with our hardware
- Work in partnership with our ML Research and Hardware Engineering teams
- Provide expertise and guidance on hardware architecture from a programmer's perspective, ensuring seamless integration with the software stack
Requirements:
- Bachelor of Computer Science or equivalent degree
- Experience optimizing software for specialized hardware, employing techniques such as parallelism, SIMD programming, C, assembly-level optimization, or GPU/CUDA programming
- Language: at least one of assembly, C++, C, Zig, or Rust
Preferences :
- Experience implementing kernels for ML models such as Transformers
- Experience using and implementing distributed parallelism techniques such as AllReduce, AllToAll, data parallelism, tensor parallelism.
- Familiarity with how compilers work
Compensation: The US base salary for this full-time position is $120,000 - $400,000 + equity + benefits
As part of our dedication to the diversity of our team and our focus on creating an inviting and inclusive work experience, MatX is committed to a policy of Equal Employment Opportunity and will not discriminate against an applicant or employee on the basis of race, color, religion, creed, national origin or ancestry, sex, gender, gender identity, gender expression, sexual orientation, age, physical or mental disability, medical condition, marital/domestic partner status, military and veteran status, genetic information or any other legally recognized protected basis under federal, state or local laws, regulations or ordinances.
All candidates must be authorized to work in the United States and work from our offices in Mountain View Tuesdays-Thursdays.
This position requires access to information that is subject to U.S. export controls. This offer of employment is contingent upon the applicants capacity to perform job functions in compliance with U.S. export control laws without obtaining a license from U.S. export control authorities.
Recommended Jobs
Backend Software Engineer - Mobile
At Perplexity, we've experienced tremendous growth and adoption since publicly launching the world's first fully functional conversational answer engine in 2022. We've grown from answering 2.5 millio…
Customer Success Analyst
ONSITE SAN FRANCISCO, CA. NOT A REMOTE POSITION. About Us: CuraeSoft is dedicated to empowering businesses and individuals with innovative and intuitive software solutions that drive growth, effi…
IT Technician
DESCRIPTION OF POSITION This job description is a record of the essential functions of the listed job. The job description provides the employee, CEO, Human Resources, applicants, and other agenci…
Senior Product Manager, Catalog
Since we opened our doors in 2009, the world of commerce has evolved immensely, and so has Square. After enabling anyone to take payments and never miss a sale, we saw sellers stymied by disparate, o…
Prep Cook (AM)
Prep Cook (AM) Location Los Angeles, CA (West Adams area) : Description: WHY HOMESTATE// Restaurants are fun (it's like throwing a party everyday)! Hourly rate starting at $21 Refer a friend, g…
Senior · Staff · Principal Frontend Engineer
Senior / Staff / Principal Frontend Engineer Location: Onsite San Francisco We have multiple startups interested in talent. Here is a generic summary. Instead of a perfect job description, we pre…
Anesthesiologist
A private Anesthesia Group in beautiful Southern California is interviewing for an open Anesthesiologist role with a partnership track! This group is going to the next level and seeking the right pro…
Senior Manager, Studio Operations
OVERVIEW OF THE COMPANY Fox Sports FOX Sports, the industry leader in live events, is the umbrella entity representing Fox Corporation’s wide array of multi-platform US-based sports assets. Bui…
Software developer
, 20 hours per week Position: Software Developer Company Overview: Aniesispharma is a leading pharmaceutical company based in California, dedicated to developing innovative treatments for …
Strategic Customer Success Manager
Sales at Brex Sales is the growth engine at Brex. We bring in new customers, expand existing relationships, and drive the company’s bottom line. With unlimited territories and uncapped opportunity…