Sr. Staff Software Engineer

Hyderabad, Telangana
Permanent
Full-time

2 months ago

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 SKILLSHands on experience in C/CPP, Python, NumPy,openCLC++Hand on experience in ML frameworks like TensorFlow, PyTorch and ONNXHand on experience in TVMBlaize 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.

Blaize

Apply Now