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