最佳PyTorch工具集合：AI开发者的实用资源
cs224u
CS224u自然语言理解
斯坦福大学自然语言理解课程代码库与学习资源
lucent
LucentPyTorch
将PyTorch神经网络可视化和解释的开源库
schedule_free
Schedule-Free Learning优 化器
自适应训练优化器，无需预设停止时间
pytorch_memlab
pytorch_memlabCUDA
PyTorch CUDA内存分析与优化工具
Captum
AI工具Captum
开源PyTorch模型可解释性分析工具库Captum
Trace
Trace自动微分
创新AutoDiff工具助力AI系统端到端训练
AQLM
AQLM大语言模型
加性量化技术实现大型语言模型高效压缩
EfficientQAT
EfficientQAT大语言模型
高效量化训练技术助力大型语言模型压缩
RWKV_Pytorch
RWKVPyTorch
RWKV大语言模型的纯PyTorch推理框架
SRGAN-PyTorch
SRGAN超分辨率
基于GAN的单图像超分辨率实现
nanoT5
T5模型预训练
轻量高效的T5模型训练框架
sheeprl
SheepRL强化学习
基于PyTorch的强化学习框架支持多种算法和环境
flash-attention
FlashAttentionCUDA
高效注意力机制加速深度学习模型训练
poutyne
PoutynePyTorch
简化PyTorch开发 加速神经网络训练
nerf-factory
NeRFPyTorch
PyTorch实现的NeRF算法集合
HolisticTraceAnalysis
性能分析分布式训练
高效分析分布式训练性能瓶颈的开源工具
DI-hpc
DI-HPC强化学习
高性能计算组件加速强化学习算法
Tensor-Puzzles
张量编程PyTorch
21个张量编程挑战助力深入理解PyTorch和NumPy
d2l-ai-solutions-manual
动手学深度学习习题解答
《动手学深度学习》习题解答与代码实现
HashNeRF-pytorch
NeRFPyTorch
纯PyTorch实现的高速NeRF训练框架
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号