Job Description

Job Title- PyTorch & Python ML Engineer

Location- Mumbai/Bangalore/Hyderabad/Gurgaon/Indore

Mode- Onsite

Job Type: Full-Time


Core Responsibilities

The role designs and owns end-to-end ML pipelines implemented in Python, covering audio data ingestion, preprocessing, feature extraction, training invocation, evaluation, and artifact generation.

The role is responsible for implementing and maintaining large-scale speech models using PyTorch, including transformer-based architectures for speech-to-text and text-to-speech, ensuring correctness, performance, and extensibility.

The role owns audio data ETL and dataset management workflows, including data validation, labeling interfaces, dataset versioning, and traceability between datasets and trained models.

The role ensures that model implementations are training- and inference-compatible, avoiding mismatches in tensor shapes, precision, preproce...

Apply for this Position

Ready to join Confidential? Click the button below to submit your application.

Submit Application