AI Engineer Model Optimization & Acceleration

Sunrise Biztech Systems; Bangalore Rural; On-site; INR 1,200,000 - 2,400,000; Full time

Job summary

Sunrise Biztech Systems in Bengaluru, Karnataka is hiring an AI Engineer to optimize and deploy ML models across CPU, GPU and NPU platforms. You will drive production-ready AI systems for robotics, healthcare, and automotive domains.

The role emphasizes performance tuning, model quantization, and cross-framework deployment using PyTorch, ONNX, and CUDA. You will work on latency reduction, memory management and scalable inference pipelines.

Qualifications

• Strong experience with PyTorch and ONNX
• Proficient in Python and C++
• Experience with GPU/hardware acceleration (CUDA, ROCm)
• Solid understanding of transformers and CNNs
• Knowledge of optimization and quantization for performance

Responsibilities

• Optimize models including generative, vision, and multi-modal networks
• Port models across frameworks and runtimes
• Deploy on hardware accelerators and optimize throughput
• Improve inference latency, memory usage, and batching
• Profile, debug, and tune system and model performance

Skills

PyTorch

ONNX

Python

C++

GPU acceleration

Tools

CUDA

ONNX Runtime

Job description

Role Overview

Seeking an AI Engineer to optimize and deploy ML models across heterogeneous platforms (CPU, GPU, NPU).
Work on scalable, production-ready AI systems across domains like robotics, healthcare, and automotive.

Key Responsibilities

• Optimize diverse models: generative (LLMs, diffusion), vision (classification, detection, segmentation), CPU,NPU,GPU,AI,multi-modal,speech,LLMS,PyTorch,ONNX,runtimes,GNU,NPU,FP32,debug,lower precision
• Port models across frameworks (e.g., PyTorch ONNX runtimes)
• Deploy on hardware accelerators (GPU/NPU) and optimize performance
• Improve inference latency, throughput, and memory (batching, caching, parallelism, fusion)
• Apply quantization and model compression (FP32 lower precision)
• Profile and debug system and model performance

Required Skills

• Strong in PyTorch (or similar), ONNX (or equivalent)
• Proficient in Python and C++
• Experience with GPU/hardware acceleration (CUDA,ROCm or similar)
• Solid understanding of deep learning models (transformers, CNNs)
• Knowledge of optimization, quantization, and performance tuning

Good to Have

• Edge AI or embedded deployment
• Generative or multi-modal AI systems
• Distributed inference or streaming pipelines