Description

You will architect, productize, and maintain scalable Python-based AI/ML codebases, converting research into production-ready services. This role is on-site.

Responsibilities

• Design modular architectures and clean APIs for ML systems.
• Optimize and debug GPU workloads using PyTorch and CUDA for training and inference.
• Build and maintain unit, integration, and regression test suites with CI/CD integration.
• Mentor peers and lead design reviews.

Required Skills

• 5+ years of hands-on Python development and software design.
• Deep expertise in PyTorch, CUDA, and GPU programming.
• Strong testing experience with pytest, integration, and regression testing.
• Familiarity with CI/CD pipelines and clean coding practices.
• Experience with Jenkins, Docker, and Kubernetes.
• Bachelor's degree in Computer Science or related field.

Preferred Skills

• Computer Vision experience (Detection, Segmentation, Tracking).
• Experience with TorchScript, ONNX, or TensorRT.
• Exposure to GenAI coding tools like Claude Code or GitHub Copilot.

Key Skills

Education

Bachelor's degree