Job Description

About Blaize

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 edge and in the data center. Blaize delivers real-time insights at low power, high efficiency, and minimal cost--driving next-gen AI adoption across industrial, automotive, and smart infrastructure markets.
With $330M+ raised from investors including DENSO, Mercedes-Benz AG, Samsung, and Franklin Templeton, Blaize is a global team headquartered in California with branches in India, the UK, Middle East--and now growing its strategic presence in East Asia.

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.