Job Description

Blaize provides a full-stack programmable processor architecture suite and low-code/no-code software platform that enables AI processing solutions for high-performance computing at the network's edge and in the data center. Blaize solutions deliver real-time insights and decision-making capabilities at low power consumption, high efficiency, minimal size and low cost. Blaize has raised over $330 million from strategic investors such as DENSO, Mercedes-Benz AG, Magna, and Samsung and financial investors such as Franklin Templeton, Temasek, GGV, Bess Ventures, BurTech LP LLC, Rizvi Traverse, and Ava Investors. Headquartered in El Dorado Hills (Calif.), Blaize has more than 200 employees worldwide with teams in San Jose (Calif.) and Cary (N.C.), and subsidiaries in Hyderabad (India), Leeds and Kings Langley (UK), and Abu Dhabi (UAE).
JOB RESPONSIBILITIES

• Bring up, test and debug neural networks using ML frameworks like TensorFlow, PyTorch, ONNX etc. for Blaize-GSP.
• Bring up and enhance TVM features

• Analyze and enhance efficiency & stability of neural networks.

• Develop & maintain Model Conversion Tool software stack for Blaize-GSP.
• Network Optimization, Node fusion, graph lowering, adding custom operations, profiling & performance fine tuning.
• Model Quantization

EDUCATION AND EXPERIENCE

• MS or BS in computer science or Electronics
• 8-12yrs years of relevant experience or an equivalent combination of experience and education.

REQUIRED KNOWLEDGE, SKILLS, AND ABILITIES

• Hands on experience in C,C++,Python, NumPy, ML frameworks like TensorFlow, PyTorch, ONNX and others.
• Good knowledge of Linear algebra
• Knowledge of NWs optimization, graph lowering and finetuning
• Good analytical skills
• Good understanding of algorithms, OOPS concepts and SW Design Patterns.
• Good debugging skills.
• Strong knowledge of TVM FW
• Experience or knowledge in HW Architecture is an added advantage.
• Experience on full stack framework development like any of Multimedia frameworks, GStreamer, OpenVx, OpenMax, OpenGL, OpenGL-ES, Vulkun, Mesa, etc. is a plus.
• Experience on driver development on Linux platform.

• CLC/assembly compute kernels for the Blaize-GSP.

MANDATORY SKILLS
Hands on experience in C/CPP, Python, NumPy,openCLC++
Hand on experience in ML frameworks like TensorFlow, PyTorch and ONNX
Hand on experience in TVM
Blaize is an equal opportunity employer. We pride ourselves on having a diverse workforce and we do not discriminate against any employee or applicant because of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition, or any other basis protected by law. We respect the gender, gender identity, and gender expression of our applicants and employees, and we honor requests for preferred pronouns. It is our policy to comply with all applicable national, state, and local laws pertaining to nondiscrimination and equal opportunity.