深度学习

网站技能导入器展示的深度学习技能与工作流。

11 個技能
A
pytorch-patterns

作者 affaan-m

pytorch-patterns 可協助你以裝置無關的模式、可重現的實驗,以及明確的 tensor 處理來撰寫、審閱與除錯 PyTorch 程式碼。這個 pytorch-patterns 技能適合用來打造更乾淨的訓練迴圈、重構模型,以及取得實用的 PyTorch 指引。

程式碼编辑
收藏 0GitHub 156.2k
K
torchdrug

作者 K-Dense-AI

torchdrug 是一套以 PyTorch 為核心的分子與蛋白質機器學習工具箱。你可以透過 torchdrug 技能來選擇任務、資料集與模組化模型,涵蓋圖神經網路、蛋白質建模、知識圖譜推理、分子生成與逆合成。它最適合客製化模型開發與可重現的設定,而不只是現成示範。

Machine Learning
收藏 0GitHub 21.4k
K
torch-geometric

作者 K-Dense-AI

適用於 PyTorch Geometric 圖神經網路的 torch-geometric 技能指南。可用來取得 torch-geometric 安裝協助、torch-geometric 使用方式、圖分類、節點分類、連結預測、異質圖、自訂 MessagePassing 層,以及為 Machine Learning 工作流程擴展 GNN。

Machine Learning
收藏 0GitHub 21.4k
H
huggingface-vision-trainer

作者 huggingface

huggingface-vision-trainer 可協助你安裝並使用一套 Hugging Face 視覺訓練技能,支援物件偵測、影像分類,以及 SAM/SAM2 分割等訓練工作。內容涵蓋資料集準備、雲端 GPU 環境設定、評估、Trackio 記錄與將結果推送到 Hub。很適合後端自動化與可重複的訓練流程。

後端开发
收藏 0GitHub 10.4k
H
huggingface-llm-trainer

作者 huggingface

huggingface-llm-trainer 可協助你在 Hugging Face Jobs 上使用 TRL 或 Unsloth 訓練或微調語言與視覺模型。這個 huggingface-llm-trainer 技能適用於 SFT、DPO、GRPO、reward modeling、資料集檢查、GPU 選擇、Hub 儲存、Trackio 監控,以及供後端開發流程使用的 GGUF 匯出。

後端开发
收藏 0GitHub 10.4k
H
huggingface-best

作者 huggingface

huggingface-best 技能可透過檢查 Hugging Face 基準排行榜,並依裝置限制與模型大小篩選,幫你找到某項任務的最佳模型。當你需要的是實用的候選清單,而不是泛用模型列表時,它很適合用在程式撰寫、推理、聊天、OCR、RAG、語音、視覺或多模態工作上的模型推薦。

模型評測
收藏 0GitHub 10.4k
K
transformers

作者 K-Dense-AI

這個 transformers 技能可協助你使用 Hugging Face Transformers 進行模型載入、推論、tokenization 與 fine-tuning。它是一份實用的 transformers 指南,適用於 Machine Learning 工作,涵蓋文字、視覺、音訊與多模態流程,並提供從快速 baseline 到自訂訓練的清楚路徑。

Machine Learning
收藏 0GitHub 0
K
scvi-tools

作者 K-Dense-AI

scvi-tools 是一個用於機率式單細胞分析的 Python 框架。這個 scvi-tools 技能可用於批次校正、潛在嵌入、帶不確定性估計的差異表現分析、遷移學習,以及多模態整合。它特別適合單細胞 RNA-seq、ATAC、CITE-seq、multiome 與空間流程,尤其是進階 Machine Learning 使用情境。

Machine Learning
收藏 0GitHub 0
K
pytorch-lightning

作者 K-Dense-AI

這個 pytorch-lightning 技能可協助你用 LightningModules 和 Trainer 來整理 PyTorch 專案。這份 pytorch-lightning 指南適合用來了解安裝、訓練、驗證、記錄、檢查點,以及跨多 GPU 或 TPU 工作流程的分散式執行。

後端开发
收藏 0GitHub 0
K
pyhealth

作者 K-Dense-AI

pyhealth 可協助你建立臨床與醫療深度學習流程,採用 Dataset → Task → Model → Trainer → Metrics 的工作流。這個 pyhealth 技能適用於 MIMIC-III/IV、eICU、OMOP、SleepEDF、ChestXray14、EHRShot,以及預測、藥物推薦、睡眠分期、ICD 編碼、EEG 事件與醫療代碼對照等任務。

Scientific
收藏 0GitHub 0
K
pufferlib

作者 K-Dense-AI

pufferlib 是一套高效能強化學習技能,適合快速平行模擬、向量化 rollout 與多智能體訓練。這份 pufferlib 指南可協助你安裝、理解 pufferlib 的用法,並將 RL 流程延伸到 Gymnasium、PettingZoo、Atari、Procgen 或 NetHack 風格環境。特別適合著重吞吐量與可擴充 PPO 工作流程的程式生成需求。

程式碼生成
收藏 0GitHub 0
深度学习