CareerPlan

SDE II - Gen AI

BangaloreFull-time2026-10-07 → 2026-11-10

Core

Engineer responsible for end-to-end deployment of generative AI models (image generation, VTON) to iOS and Android devices, optimizing for latency, memory, and battery constraints.

Role type

Senior IC machine-learning engineer (on-device generative AI)

Builds

Production-ready model pipelines for image generation and personalization on consumer mobile devices

Domain

Mobile computing + Generative AI

Deliverable

production ML models

Required skills

Python, PyTorch, Computer Vision, Generative AI, LoRA, Model optimization, Quantization, Inference runtimes (ONNX, TensorRT, Core ML, TFLite), Debugging, Profiling

Preferred skills

FLUX, Stable Diffusion, SDXL, ControlNet, CUDA, Metal, Android NNAPI, C++, Distributed training, Model compression

Technologies

PyTorch, ONNX, TensorRT, Core ML, TFLite, ONNX Runtime, CUDA, Metal, NNAPI

Responsibilities

Train and fine-tune image-generation models using LoRA and adapter techniques; Optimize model inference for on-device constraints (latency, memory, thermal); Convert and deploy models using ONNX/TensorRT/Core ML/TFLite; Profile and debug performance across CPU/GPU/NPU; Build production pipelines from training to device deployment.

Seniority

Mid-level, hands-on IC