FIND_THE_RIGHTJOB.

Software Engineer

Menlo Park, United States

Payrate: $78.00 - $96.30/hr.

Summary:
The AI and Systems Co-Design team has a mission to explore, develop, and help produce high-performance software and hardware technologies for AI. Our team defines and drives the AI software and hardware roadmap. We are seeking a candidate who will work on a foundational tool for our internal workloads on current and next-generation AI platforms. Specifically, this position focuses on collecting, processing, storing, and analyzing various operators and workloads.

Responsibilities:

Extract operators (e.g. aten, triton) from AI/ML models.
Run operators on multiple devices and collect performance data.
Process collected data and stored it to a database while maintaining data integrity.
Implement, improve, and maintain programmatic and web interfaces to query and analyze performance data stored in the database.
Collaborate as part of a project team to coordinate development and determine project scope and limitations.
Review project requests to estimate time and cost required to complete the project.

Must-have skills:

Hands-on experience with product-level Python programming
Proficiency in PyTorch, Kineto trace, dispatcher
Hands-on Experience with CUDA, Triton kernels
Hands-on experience in database management and SQL
Proficiency in Linux and Bash
Ability to work independently

Good to have skills:

Experience in LLM especially Llama.
Knowledge of CI-based testing and automation
Education/Experience
At least three years of experience with above-mentioned skills is required for this role.

Pay Transparency: The typical base pay for this role across the U.S. is: $78.00 - $96.30/hr. Final offer amounts, within the base pay set forth above, are determined by factors including your relevant skills, education and experience and the benefits package you select. Full-time employees are eligible to select from different benefits packages. Packages may include medical, dental, and vision benefits, 10 paid days off, 401(k) plan participation, commuter benefits and life and disability insurance.

For information about our collection, use, and disclosure of applicant's personal information as well as applicants' rights over their personal information, please see our Privacy Policy ( https://www.aditiconsulting.com/privacy-policy ).

Aditi Consulting LLC uses AI technology to engage candidates during the sourcing process. AI technology is used to gather data only and does not replace human-based decision making in employment decisions. By applying for this position, you agree to Aditi’s use of AI technology, including calls from an AI Voice Recruiter.

#AditiConsulting
#25-22272

Similar jobs