Software Engineer II

Blaize; Hyderabad; On-site; INR 1,200,000 - 1,800,000; Full time

Job summary

A leading AI technology company in Hyderabad is seeking a Software Engineer – II to enhance the SDK within the performance library team. The role involves designing and optimizing high-performance execution for innovative hardware while collaborating across teams such as compiler and machine learning. The ideal candidate has substantial experience in software engineering focused on performance, strong C and C++ skills, and familiarity with ONNX or PyTorch. This position offers a dynamic working environment to innovate at the forefront of AI technology.

Qualifications

• 5–8 years of hands-on software engineering experience in performance-critical systems.
• Strong proficiency in C and C++, including extensive use of STL libraries.
• Solid experience with ONNX and/or PyTorch operators.

Responsibilities

• Design, implement, and maintain operators and kernels within the Blaize SDK.
• Optimize operator performance for efficiency across workloads.
• Collaborate with cross-functional teams including hardware, compiler, and ML engineers.

Skills

C

C++

Data Structures and Algorithms

STL Libraries

ONNX / PyTorch Operators

Graph and Node Optimization

Education

Bachelor’s or Master’s degree in Computer Science

Job description

About Blaize

Blaize is building a hybrid AI platform engineered to support edge-to-cloud intelligence at scale—delivering efficient, scalable AI designed for complex, multimodal workloads across industries. We serve critical infrastructure sectors including smart city, defense, retail, manufacturing, healthcare, and automotive.

Location

Hyderabad (India)

Description

Blaize is seeking a Software Engineer – II to join the SDK – DNN Performance library team. In this role, you will design, develop, and optimize operators and kernels within the Blaize SDK, enabling high-performance execution on current and next-generation Blaize hardware. You will work closely with compiler, hardware, and ML teams to deliver efficient and scalable solutions.

Responsibilities

• Design, implement, and maintain operators and kernels within the Blaize SDK and perlib.
• Optimize operator performance and improve execution efficiency across workloads.
• Enable and support new features and performance improvements for next-generation Blaize chips.
• Collaborate with cross-functional teams including hardware, compiler, and ML engineers.
• Analyze performance bottlenecks and implement optimizations at the graph and kernel levels.

Education & Experience

• Bachelor’s or master’s degree in computer science or a related field.
• 5–8 years of hands‑on software engineering experience, preferably in performance‑critical systems.
• Strong proficiency in C and C++, including extensive use of STL libraries.
• Solid experience with ONNX and/or PyTorch operators, including graph and node‑level optimization.
• Experience writing parallel kernels for GPUs or similar accelerator architectures.
• Understanding performance optimization techniques for compute‑intensive workloads.
• Basic knowledge of machine learning networks and large language models (LLMs) is a plus.

Mandatory Skills

• C, C++
• Data Structures and Algorithms
• STL Libraries
• ONNX / PyTorch Operators
• Graph and Node Optimization

Equal Opportunity Employer

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 honour requests for preferred pronouns. It is our policy to comply with all applicable national, state, and local laws pertaining to non-discrimination and equal opportunity.