08 Oct
|
Important Group
|
Chennai
08 Oct
Important Group
Chennai
We are looking
for a Senior Software Engineer to develop and optimize deep learning models,
including CNNs, LLMs, and MoE, for efficient inference across CPU, GPU,
hardware accelerators, and edge devices. The role focuses on quantization,
model compression, high-performance kernel implementation, transformer
optimization, and production deployment.
Responsibilities:
• Develop and optimize deep learning models (CNNs,
LLMs, MoE) for efficient inference across CPU, GPU, hardware accelerators, and
edge devices.
• Design and implement quantization algorithms
(PTQ, QAT, GPTQ, AWQ) from scratch.
• Apply model compression techniques such as
pruning, decomposition, and distillation.
• Implement and optimize quantized kernels (INT8,
INT4, FP8) using C++ for high performance.
• Translate research papers into production-ready
implementations.
• Optimize latency, throughput, and memory usage
for real-world deployment.
• Work on transformer optimization including
KV-cache, PEFT (LoRA/QLoRA), and MoE models.
• Profile, benchmark, and debug model performance
across different hardware platforms.
• Collaborate with ML, compiler, and hardware
teams to deliver optimized solutions.
Requirements
Education:
BE/BTech/MS/MTech in Computer Science or a related field.
Technical Skills (Must haves):
• 4+ years of relevant experience.
• Strong programming skills in Python and C++.
• Proven experience in quantization algorithms
(PTQ, QAT, GPTQ, AWQ).
• Hands-on experience in pruning, model
compression, and inference optimization.
• Experience implementing quantization or
optimization techniques from scratch.
• Strong understanding of CNNs, Transformers, and
LLM architectures.
• Experience with PyTorch / ONNX and model
deployment pipelines.
• Solid problem-solving and performance
optimization skills.
Need to have (Can be bridged):
No
additional bridged skills were specified.
Good to have (Not essential):
• Experience with MoE architectures and PEFT
techniques (LoRA, QLoRA).
• Knowledge of TensorRT, ONNX Runtime, TVM, and
MLIR.
• Familiarity with hardware-aware optimization
across GPU, NPU, and edge devices.
• Experience in research paper implementation or
open-source contributions.
Preferred Qualifications (Optional):
No
additional preferred qualifications were specified.
📌 SSE - Optimization Engineer (Chennai)
🏢 Important Group
📍 Chennai