About the role
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
- Good 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
If you enjoy solving problems like “How do we make a large generative model run fast, efficiently, and reliably on a phone?”, this role should be exciting.