Deep Learning Collaborating Pytorch
• Fine-tune pretrained deep learning models (CNN/Transformer) on task-specific datasets (e.g., classification, detection, segmentation). • Apply model compression techniques: quantization (PTQ/QAT), pruning, knowledge distillation, etc. • Convert models to deployment-ready formats (ONNX, TensorRT, OpenVINO...). • Optimize model inference for edge deployment: latency, FPS, throughput, memory footprint. • Benchmark performance on different platforms: Jetson, ARM, Qualcomm SNPE, OpenVINO, or AI SDKs. • Collaborate with the software and hardware teams to validate models in production environments. • Maintain and improve MLOps pipeline: training logs, version control, artifact tracking.
• Bachelor's degree or higher in Computer Science, Data Science, EE, or related fields. • 2+ years of experience in deep learning (PyTorch or TensorFlow). • Hands-on experience with model optimization techniques and deployment. • Familiar with ONNX, TFLite, TensorRT, or similar deployment frameworks. • Understanding of inference performance metrics (latency, FLOPs, memory usage).
• Benefits will be shared in detail with successful candidates.