深度学习最佳AI工具集合:网站、应用和开源项目

pytorch_tabular

pytorch_tabular

PyTorchPyTorch Lightning

表格数据深度学习的简易化解决方案

mup

mup

Maximal Update ParametrizationMuTransfer

大规模神经网络的稳定超参数优化方法

torchdyn

torchdyn

TorchdynPyTorch

PyTorch数值深度学习库,支持微分方程和数值方法

DeepLearing-Interview-Awesome-2024

DeepLearing-Interview-Awesome-2024

DeepLearning-Interview-Awesome-2024大模型

2024深度学习面试题与资源汇总

pytorch-toolbelt

pytorch-toolbelt

PyTorch深度学习

专为PyTorch设计的Python库,提供高效研发和Kaggle竞赛所需的工具集

deepmd-kit

deepmd-kit

DeePMD-kit深度学习

深度学习驱动的原子势能建模与分子动力学模拟工具

vocal-remover

vocal-remover

vocal-remover深度学习

使用深度学习的开源伴奏提取工具

ICON

ICON

ICONCVPR 2022

利用法线预测实现穿衣人体的隐含表达,从RGB图像生成高细节的3D模型,包括衣服和人体网格

TransformerEngine

TransformerEngine

Transformer EngineNVIDIA

用于在 NVIDIA GPU 上加速 Transformer 模型的库

lightning-flash

lightning-flash

lightning-flashPyTorch

跨数据领域和任务的AI模型训练与处理解决方案

benchmark_VAE

benchmark_VAE

pythae变分自编码器

统一实现常见变分自编码器并提供基准比较

DDSP-SVC

DDSP-SVC

DDSP-SVC声音转换

低硬件需求语音转换工具,支持多音色训练

petastorm

petastorm

Petastorm深度学习

开源数据访问库,支持单机或分布式训练和评估深度学习模型,直接从Apache Parquet格式数据集中读取数据

rainbow-is-all-you-need

rainbow-is-all-you-need

RainbowDQN

从DQN到Rainbow的深度强化学习方法

nitrain

nitrain

Nitrain医学影像

医学图像采样与增强的跨平台Python库

hivemind

hivemind

Hivemind深度学习

去中心化的分布式深度学习PyTorch库

Anime2Sketch

Anime2Sketch

Anime2Sketch图像处理

动漫、插画和漫画图像的草图提取工具

labml

labml

LabML深度学习

通过手机监控深度学习模型训练和硬件使用情况

torchio

torchio

TorchIO深度学习

深度学习医疗图像处理工具集

yolor

yolor

YOLORYOLOv4

改进的多任务统一网络实时对象检测模型