Know ATS Score
CV/Résumé Score
  • Expertini Resume Scoring: Our Semantic Matching Algorithm evaluates your CV/Résumé before you apply for this job role: Machine Learning Software Engineer Python/PyTorch.
India Jobs Expertini

Urgent! Machine Learning Software Engineer - Python/PyTorch Job Opening In Hyderabad – Now Hiring Stealth Mode Startup - AI Product Based Company

Machine Learning Software Engineer Python/PyTorch



Job description

<p>Job Summary : <br/><br/>We are seeking a highly experienced and skilled Machine Learning Software Engineer with 8-10 years of experience to join our team.

The ideal candidate will be a deep learning expert with a strong background in optimizing and deploying machine learning models on specialized hardware, particularly ML accelerators.

</p><p><br/></p><p>This role is critical for bridging the gap between theoretical model development and practical, high-performance inference on target platforms.

A key focus of this position will be on model quantization and other optimization techniques to maximize efficiency and performance.

<br/><br/>Key Responsibilities : <br/><br/>- Model Porting & Deployment : Port and deploy complex deep learning models from various frameworks (e.g., PyTorch, TensorFlow) to proprietary or commercial ML accelerator hardware platforms (e.g., TPUs, NPUs, GPUs).<br/><br/>- Performance Optimization : Analyze and optimize the performance of ML models for target hardware, focusing on latency, throughput, and power consumption.<br/><br/>- Quantization : Lead the efforts in model quantization (e.g., INT8, FP16) to reduce model size and accelerate inference while preserving model accuracy.<br/><br/>- Profiling & Debugging : Utilize profiling tools to identify performance bottlenecks and debug issues in the ML inference pipeline on the accelerator<br/><br/>- Collaboration : Work closely with the ML research, hardware, & software teams to understand model requirements and hardware capabilities, providing feedback to improve both.<br/><br/>- Tooling & Automation : Develop and maintain tools and scripts to automate the model porting, quantization, and performance testing workflows<br/><br/>- Research & Innovation : Stay current with the latest trends and research in ML hardware, model compression, and optimization techniques.<br/><br/>Experience : <br/><br/>- 8-10 years of professional experience in machine learning engineering, with a focus on model deployment and optimization.<br/><br/>Technical Skills : <br/><br/>- Deep expertise in deep learning frameworks such as PyTorch and TensorFlow.<br/><br/>- Proven experience in optimizing models for inference on GPUs, NPUs, TPUs, or other specialized accelerators<br/><br/>- Extensive hands-on experience with model quantization (e.g., Post-Training Quantization, Quantization-Aware Training).<br/><br/>- Strong proficiency in C++ and Python, with experience writing highperformance, low-level code<br/><br/>- Experience with GPU programming models like CUDA/cuDNN<br/><br/>- Familiarity with ML inference engines and runtimes (e.g., TensorRT, OpenVINO, TensorFlow Lite).<br/><br/>- Strong understanding of computer architecture principles, including memory hierarchies, SIMD/vectorization, and cache optimization<br/><br/>- Version Control : Proficient with Git and collaborative development workflows<br/><br/>- Education : Bachelor's or Master's degree in Computer Science, Electrical Engineering, or a related field.<br/><br/>Preferred Qualifications :<br/><br/>- Experience with hardware-aware model design and co-design.<br/><br/>- Knowledge of compiler technologies for deep learning.<br/><br/>- Contributions to open-source ML optimization projects.<br/><br/>- Experience with real-time or embedded systems.<br/><br/>- Knowledge of cloud platforms (AWS, GCP, Azure) and MLOps best practices.<br/><br/>- Familiarity with CI/CD pipelines and automated testing for ML models<br/><br/>- Domain knowledge in areas like computer vision, natural language processing, or speech recognition.</p> (ref:hirist.tech)


Required Skill Profession

Computer Occupations



Your Complete Job Search Toolkit

✨ Smart • Intelligent • Private • Secure

Start Using Our Tools

Join thousands of professionals who've advanced their careers with our platform

Rate or Report This Job
If you feel this job is inaccurate or spam kindly report to us using below form.
Please Note: This is NOT a job application form.


    Unlock Your Machine Learning Potential: Insight & Career Growth Guide