
TorchCode
duoan
🔥 LeetCode for PyTorch — practice implementing softmax, attention, GPT-2 and more from scratch with instant auto-grading. Jupyter-based, self-hosted or try online.
AI 简介
TorchCode 是一个面向机器学习工程师的 PyTorch 实操训练平台,旨在通过从零实现核心算子与模型提升面试实战能力。项目提供 40 道精选题目(如 softmax、LayerNorm、MultiHeadAttention、GPT-2 模块等),基于 Jupyter Notebook 构建,支持本地自托管或在线即用;内置自动评测系统,可验证前向/反向传播正确性、梯度一致性与性能,并提供分步提示、参考解与进度追踪。适用于 ML 工程师准备技术面试、巩固张量编程基础及深入理解深度学习底层实现。
Jupyter Notebook
4.3k
Stars
377
Forks
9
Watchers
8
Issues
Star 增长
今日0
近 7 天0
近 30 天+101
综合评分66.73
默认分支master