搜索结果:gpus(6 个)
按分类缩小范围
ml-engineering
综合 71
机器学习工程开源电子书。
TensorRT-LLM
综合 70
提供易用的 Python API,针对 NVIDIA GPU 优化 LLM 高效推理加速。
pytorch-lightning
综合 49
无需修改代码,即可在单卡或上万张 GPU 上预训练及微调任意规模的 AI 模型。
cudf-spark
综合 46
cudf-spark(原 RAPIDS Accelerator for Apache Spark)是 NVIDIA 开源的 Spark 插件。它利用 cuDF 库调用 GPU 加速 Apache Spark 的大数据处理;其核心功能是在保证计算结果与原生 Spark 严格一致的同时,通过 GPU 提升 SQL 性能,并支持向 XGBoost 等 GPU 应用零拷贝传输数据。
zinc
综合 44
Zig 推理引擎,支持在 AMD GPU 和 Apple Silicon 上进行本地 LLM 推理。
TensorRT
综合 43
NVIDIA GPU 高性能深度学习推理 SDK,本仓库包含其开源组件。