Skip to content

Problems Questions Drills Roles Skills Interview Contest Leaderboard

Problems Questions Drills Roles Skills Interview Contest Leaderboard

Home
Machine Learning Engineer
Questions
PyTorch

Machine Learning Engineer

PyTorch questions

7 questions on PyTorch for machine learning engineers, easiest first.

Easy 1

Dropout after eval()easy

Medium 5

Different results every inference runmedium
Gradient of a sum of squaresmedium
Training loop without zero_gradmedium
Softmax before CrossEntropyLossmedium
Shape after a batched matmulmedium

Hard 1

view() after a transposehard

More Machine Learning Engineer skills

Classification Metrics 7 Feature Engineering 5 Training 5 Data Leakage 5 LLM Serving 5 Model Evaluation 4 Deep Learning 4 Distributed Training 4 Fine Tuning 4

PyTorch across all roles →

🏆 PyTorch leaderboard →

RED SAG Skills

Free practice for every job role: coding problems, role challenges and mock interviews. Built and managed by RED SAG Fintech Solutions.

Software Engineering

Frontend Developer interview questions
Backend Developer interview questions
Full Stack Developer interview questions
Mobile App Developer interview questions
Game Developer interview questions
Blockchain Developer interview questions

Data, AI & Machine Learning

Machine Learning Engineer interview questions
AI Engineer (LLM & Generative AI) interview questions
Data Scientist interview questions
MLOps Engineer interview questions
Prompt Engineer interview questions
Data Analyst interview questions

Cloud, DevOps & Security

DevOps Engineer interview questions
Cloud Engineer interview questions
Site Reliability Engineer interview questions
Database Administrator interview questions
Cybersecurity Analyst interview questions

© 2026 RED SAG Fintech SolutionsAll roles Coding problems Leaderboard About Careers at RED SAG