AI Engineer

Job Title: AI Engineer

Client Name:KLA India

Domain:Engineering Services & Industrial Automation

Location: India - Chennai

Salary: INR 2700000 - 3500000

Job Positions:1

Experience: 5 - 8 Years

Must Haves

Education/qualification: Preferably from premier Institute like IIT, IISC, IIIT and BITS, NIT. 

Other Tier 1 regional colleges.

-Required: MS/MTech degree with 3 or more years of Algorithm development experience.

-Required Min 3 years strong Python and PyTorch engineering skills.

-Required: Knowledge of software testing, version control, and deployment best practices.

-Preferred: Experience with data serialization and transfer protocols (e.g., gRPC, protobuf, shared memory).

Product or Semi conductor or Hardware Manufacturing company experience is a MUST. Candidates from engineering product companies have higher chances of getting shortlisted (current company or past experience)

Job stability - min 2 years in an organisation

Job Description/Preferred Qualifications

Core Responsibilities

1. Implement and maintain PyTorch-based training and inference pipelines.

2. Develop modular, reusable code for data handling, model serving, and evaluation.

3. Optimize code to run faster and keep accuracy

4. Optimize data transfer between systems (e.g., Windows ? Linux), including gRPC or other protocols.

5. Ensure code quality through testing, documentation, and CI/CD practices.

6. Support deployment of models into production environments or internal tools.

7. Collaborate with the algorithm expert to translate research into scalable solutions.

Minimum Qualifications

Qualifications/Requirements:

Required: MS/MTech degree with 5 or more years of Algorithm development experience.

Required: Strong Python and PyTorch engineering skills.

Required: Knowledge of software testing, version control, and deployment best practices.

Preferred: Experience with data serialization and transfer protocols (e.g., gRPC, protobuf, shared memory).

Preferred: Familiarity with cross-platform development and system integration.

Preferred: Math background