PyTorch

由网站技能导入器展示的 PyTorch 技能和工作流。

12 個技能
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
K
transformers

作者 K-Dense-AI

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

Machine Learning
收藏 0GitHub 0
K
stable-baselines3

作者 K-Dense-AI

這份 stable-baselines3 技能指南適用於機器學習工作流程:訓練 RL agent、串接 Gymnasium 環境,並在 PPO、SAC、DQN、TD3、DDPG 或 A2C 之間更有把握地做選擇。最適合標準的單一 agent 強化學習、快速原型開發,以及實際的 stable-baselines3 使用情境。

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
K
cellxgene-census

作者 K-Dense-AI

cellxgene-census 技能可用來以程式化方式查詢 CELLxGENE Census。適合探索表達量資料、metadata、embeddings,以及跨資料集的模式,涵蓋不同組織、疾病與細胞類型。最適合大規模單細胞分析與參考圖譜比較;如果是自己的資料,建議使用 scanpy 或 scvi-tools。

数据分析
收藏 0GitHub 0
PyTorch