We re looking for an engineer who can work across Generative AI, model fine-tuning, optimization, and on-device deployment for iOS and Android.

This role is ideal for someone who enjoys taking models beyond experimentation and making them actually work on consumer devices under real constraints such as latency, memory, battery, thermal limits, and model size .

What you ll work on

- Image Generation, VTON, identity-preserving and reference-based generation
- LoRA training, fine-tuning, personalization, and adapter-based techniques
- PyTorch-based training and experimentation
- Diffusion / Transformer-based image generation architectures
- Dataset preparation, training pipelines, and evaluation
- Model conversion and deployment using ONNX / ONNX Runtime
- Inference optimization using TensorRT
- Quantization using FP16 / INT8 / INT4 and other optimization techniques
- On-device inference on iOS using Core ML and Android using TFLite/LiteRT, ONNX Runtime, or similar runtimes
- Performance profiling and debugging across CPU, GPU, NPU / ANE
- Reducing inference latency, peak memory usage, and model footprint
- Building production-ready model pipelines from training to device deployment

What we re looking for

- 2-5 Years of Experience
- Strong experience with Python,



PyTorch, and deep learning
- Hands-on experience with Computer Vision / Generative AI
- Experience training or fine-tuning image-generation models
- Valuable understanding of LoRA, model optimization, and quantization
- Strong knowledge of inference runtimes such as ONNX, TensorRT, Core ML, TFLite/LiteRT, or ONNX Runtime
- Strong debugging, profiling, and software engineering fundamentals

Good to have

- Experience with FLUX, Stable Diffusion, SDXL, ControlNet, IP-Adapter, VAE, CLIP, or similar architectures
- CUDA / GPU optimization
- Metal / Apple Neural Engine
- Android NNAPI / GPU delegates
- C++ / Swift / Kotlin
- Distributed or multi-GPU training
- Model compression, pruning, and knowledge distillation

We re especially interested in engineers who can think end-to-end:

Data Training Fine-tuning Evaluation Quantization Model Conversion Runtime Optimization On-Device Deployment Production Monitoring

Disclaimer: This job posting has been aggregated from external source. Role details, content, and availability are subject to change. Applicants are advised to confirm the latest information directly on the company website before applying.

📌 SDE II - Gen AI (Bengaluru)
🏢 Glance
📍 Bengaluru