PyTorch

Deep learning framework of choice for research and production ML. Used for training neural networks and building ML applications.

Proficiency: Intermediate-Advanced · 2+ years

Experience

• Custom nn.Module architectures (CNNs, Transformers, MLPs)
• Training loops with Adam, learning rate scheduling
• torch.utils.data.Dataset + DataLoader pipelines
• Fine-tuning pre-trained models from HuggingFace

Related: Python, TensorFlow, ML, NumPy