lance-format/lance

GPU-enabled index training

開放

#1,080 建立於 2023年7月20日

在 GitHub 查看貢獻者指南
 (1 則留言) (0 個反應) (0 位負責人)Rust (695 個分叉)github user discovery
A-pythonPyTorchgood first issue

倉庫指標

星標
 (6,582 顆星)
PR 合併指標
 (平均合併 6天 1小時) (30 天內合併 219 個 PR)

描述

Currently the index training time is dominated by kmeans training, which can be sped up significantly via hardware acceleration.

Recently we added an API to take externally trained IVF centroids. So we could have an optional feature where the centroids are trained using pytorch or rapids then set the centroids explicitly.

Examples:

https://github.com/lancedb/lance/blob/main/python/python/lance/dataset.py#L481

貢獻者指南

研究方向
探索將 PyTorch 或 RAPIDS 整合到 Lance 中，實現 GPU 加速的 kmeans 訓練，並利用外部訓練質心 API。
技術棧
rustpythonpytorch
領域
backendmachine learningai
議題類型
功能
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-2 天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
PythonRustGPU
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
30