What are the responsibilities and job description for the Software Engineer - Kernels position at MatX?
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 hardwareWork in partnership with our ML Research and Hardware Engineering teamsProvide expertise and guidance on hardware architecture from a programmer's perspective, ensuring seamless integration with the software stackRequirements : Bachelor of Computer Science or equivalent degreeExperience optimizing software for specialized hardware, employing techniques such as parallelism, SIMD programming, C, assembly-level optimization, or GPU / CUDA programmingLanguage : at least one of assembly, C , C, Zig, or RustPreferences : Experience implementing kernels for ML models such as TransformersExperience using and implementing distributed parallelism techniques such as AllReduce, AllToAll, data parallelism, tensor parallelism.Familiarity with how compilers workCompensation : The US base salary for this full-time position is $120,000 - $400,000 equity benefitsAs 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.
Salary : $120,000 - $400,000