yoshitomo-matsubara/torchdistill
torchdistill 是一個基於 PyTorch 的工具庫,僅需撰寫 YAML 設定即可執行知識蒸餾實驗(教師-學生訓練)。提供多種現代 KD 損失、用於提取中間特徵的前向鈎子管理器,以及視覺與 NLP 任務的範例程式碼。透過 `pip install torchdistill` 安裝,並在研究中使用時引用相關論文。
opendilab/LightZero
結合蒙特卡洛樹搜尋(MCTS)與深度強化學習(DRL)的輕量級 PyTorch 工具包,旨在標準化並加速基於 MCTS 的決策演算法研究。
intelligent-machine-learning/dlrover
一個在 K8s 與 Ray 上提供自動故障容錯、自動擴縮與快速檢查點恢復的自動分散式深度學習系統,簡化大規模 AI 模型訓練。
lucidrains/alphafold3-pytorch
一個用於預測生物分子相互作用三維結構的 PyTorch 實作,並包含完整的 PDB 資料集整理工具。
strob/gentle
基於 Kaldi 構建的強大強制對齊器,可將語音音訊與文本轉錄進行對齊,提供精確的詞級時間戳。
appleboy/CodeGPT
CodeGPT 是一個基於 Go 的 CLI,使用 LLM API(OpenAI、Azure、Gemini、Anthropic、Ollama、Groq、OpenRouter)自動產生 Conventional-Commit 訊息與簡短的程式碼審查摘要。支援安裝 Git `prepare-commit-msg` 鈎子、自訂提示、語言翻譯、串流輸出與彈性設定(API 金鑰、提供者、代理、差異上下文等)。可透過 Homebrew、Chocolatey、腳本、二進位檔或 `go install` 安裝。
luciddreamer-cvlab/LucidDreamer
LucidDreamer 是一個從單張影像與文字提示中實現領域無關的 3D 高斯點陣場景生成的系統。
afc163/fanyi
結合詞典服務與 LLM 的中英命令列翻譯工具,內建免費 LLM 代理與 AI 代理整合功能。
sdv-dev/CTGAN
一組基於深度學習的合成資料生成器,能從真實的單表資料中學習,並創建高保真的合成版本。
vturrisi/solo-learn
基於 PyTorch Lightning 的自監督學習方法庫,用於無監督視覺表示學習,提供訓練與比較 SOTA 視覺模型的統一環境。
kevinhughes27/TensorKart
這是一個基於 TensorFlow 的專案,透過學習人類的遊戲錄影來訓練 AI 代理駕駛《瑪利歐賽車 64》。
codota/tabnine-vscode
在確保資料隱私與安全的前提下,提供上下文感知補全與聊天功能的 AI 程式碼助手。
NVIDIA/MatX
一個 C++20 數值運算函式庫,將類似 NumPy 的張量運算式帶入 NVIDIA GPU 與 CPU,並具備 JIT 核函式融合以達成最大效能。
kakao/khaiii
由 Kakao 開發的基於 CNN 的韓語詞彙分析器,利用機器學習將文本拆分為詞素,用於自然語言處理的預處理。
jax-md/jax-md
基於 JAX 建構的硬體加速、可微分子動力學程式庫,用於模擬材料與小尺度相互作用。
Lightning-AI/lightning-thunder
一個為 PyTorch 設計的源到源編譯器,透過 Python 風格的中間表示與可擴展的轉換外掛,優化模型在訓練與推論中的效能。
Zefan-Cai/KVCache-Factory
一個統一的實驗平台,用於實現和評估 KV 快取壓縮、檢索、合併和量化方法,以優化長上下文 LLM 推論的記憶體使用。
awaescher/OllamaSharp
一個為 Ollama API 提供全面綁定的 .NET 庫,讓 C# 開發者能輕鬆將本地與遠端 AI 模型整合至其應用程式中。
milvus-io/pymilvus
一個用於 Milvus 的 Python SDK,提供程式化介面,用於在 Milvus 向量資料庫中管理與搜尋高維向量嵌入。
ros-industrial/universal_robot
一個 ROS-Industrial 元套件,提供使用 ROS 控制與模擬 Universal Robots 硬體所需的配置與描述。
tensorflow/text
一組用於 TensorFlow 的文字處理類別與運算,可在訓練與推論的 TensorFlow 圖中實現一致的預處理。
sebp/scikit-survival
一個基於 scikit-learn 建構的 Python 模組,讓研究人員能連結共變數與事件發生時間,同時處理刪失資料。
Intuition-Lab/personal-model
適用於 macOS 的本地優先長期記憶執行環境,透過建立使用者工作模式與目標的證據連結個人模型,利用 MCP 為 AI 編碼代理提供有根據的上下文。
mapillary/inplace_abn
為 PyTorch 提供記憶體最佳化的 Batch Normalization 層,透過將 BN 與啟動函數結合為原地操作(in-place operation),在訓練期間減少高達 50% 的 GPU 記憶體使用量。
AIPexStudio/AIPex
一個在現有瀏覽器中執行的開源瀏覽器自動化代理,允許 AI 代理透過自然語言或 MCP 控制您的即時會話。
iDC-NEU/YiGraph
YiGraph 是一個基於 AAG 框架的智能圖資料分析代理系統,可將自然語言問題轉化為可執行的圖演算法工作流程,適用於複雜關係資料的分析。
Xerxes-2/clewdr
一個基於 Rust 的輕量級代理,使用會話 Cookie 將 Claude.ai 和 Claude Code 轉換為 OpenAI 相容 API 端點。
mjpost/sacrebleu
一個用於計算可共享、可比較、可重現的 BLEU 及其他翻譯指標的工具,自動化測試集管理和詞元化處理。
meta-pytorch/data
一個可擴展的 PyTorch 資料載入函式庫,引入中 epoch 檔案儲存的狀態感知載入功能,以及用於資料預處理的可組合迭代器。
NVIDIA-Merlin/Transformers4Rec
一個基於 PyTorch 的套件,將 Hugging Face Transformer 架構引入順序與會話推薦系統,支援豐富的表格輸入特徵與 GPU 加速。
machinepulse-ai/world2agent
一種開放協定與框架,透過可交換感測器系統提供結構化即時資料,標準化 AI 代理對現實世界的感知方式。
a-r-j/graphein
一個用於將生物分子結構與交互作用網絡轉換為幾何圖形表示法的 Python 函式庫,供深度學習使用。
Artelnics/opennn
用於商業、醫療與工程領域進階分析的高性能 C++ 神經網路庫。
hasktorch/hasktorch
利用 PyTorch 共享的核心 C++ 庫的 Haskell 張量與神經網路庫
spytensor/prepare_detection_dataset
一組用於在 CSV、COCO、Pascal VOC 和 Labelme 格式之間轉換物件檢測資料集標註的 Python 腳本。
HITsz-TMG/Uni-MoE
Uni-MoE 是一個基於混合專家(Mixture-of-Experts)的全模態大模型,能夠跨多種模態(包括文字、影像和音訊)進行理解和生成。
vamplabAI/sgr-agent-core
一個使用 Schema-Guided Reasoning 將結構化推理與彈性工具選擇結合的開源代理框架,用於建構研究代理。
XieZhiFa/IdCardOCR
一個適用於 Android 的離線 OCR 套件,可在無需網路連接的情況下,實現對中國身分證、駕照與護照的高速識別。
VOICEVOX/voicevox_core
為 VOICEVOX 提供核心語音合成引擎,以 C 與 Python 函式庫形式,方便於各種應用中輕鬆整合。
extropic-ai/thrml
一個用於在稀疏異質圖上高效執行區塊吉布斯採樣與基於能量模型的 JAX 庫。
huggingface/optimum-quanto
一個用於 Optimum 的 PyTorch 量化後端,透過將權重和激活轉換為 int4 或 float8 等較低精度,來減少模型大小和記憶體使用。
pepperoni21/ollama-rs
一個封裝 Ollama 本地模型 HTTP API 的 Rust crate,提供文字生成、聊天(含歷史)、串流回應、模型管理、嵌入向量,以及透過宏系統實作的 LLM 工具/函數呼叫的便捷函式。
NVIDIA-Merlin/HugeCTR
專為高效訓練與推論具有巨大嵌入表格的大規模深度學習模型而設計的 GPU 加速推薦框架。
agi-brain/xuance
一個全面且統一的深度強化學習庫,提供與 PyTorch、TensorFlow 與 MindSpore 相容的模組化 DRL 與 MARL 演算法集合。
apache/systemds
Apache SystemDS 是一個開源機器學習系統,使用高階語言管理從數據準備到模型服務的端到端數據科學生命週期。
libxsmm/libxsmm
一個使用 JIT 程式碼生成來優化跨多種 CPU 架構的深度學習與 HPC 工作負載的高性能函式庫,專為小型密集與稀疏線性代數運算設計。
kadirnar/segment-anything-video
Segment Anything Model (SAM) 的封裝版本,支援透過 pip 簡易安裝,並擴展影片分割與雲端 GPU 集成功能。
neka-nat/probreg
一種使用機率模型的點群配準程式庫,為剛性與非剛性變換提供了比ICP更穩健的替代方案。