torch

共找到 34 个相关项目

flashinfer-ai/flashinfer

Python

FlashInfer：大语言模型服务的核心计算库

attention cuda distributed-inference
flashinfer-ai
flashinfer-ai 开发者
4.8k
670
98
+14
排名 #7
1月27日
查看详情

dusty-nv/jetson-containers

Jupyter Notebook

适用于NVIDIA Jetson和JetPack-L4T的机器学习容器

containers docker dockerfiles
dusty-nv
dusty-nv 开发者
4.4k
795
70
+79
排名 #19
2月16日
查看详情

iree-org/iree

C++

一个基于MLIR的可重定向机器学习编译器与运行时工具包。

compiler cuda jax
iree-org
iree-org 开发者
3.6k
839
145
+5
排名 #8
2月13日
查看详情

breezedeus/Pix2Text

Jupyter Notebook

一款基于SMALL模型的开源Python3工具，能够识别图像中的版面布局、表格、数学公式以及文本，并将其转换为Markdown格式。作为Mathpix的免费替代方案，它实现了视觉内容到文本表征的无缝转换，并支持80多种语言。

image-to-markdown latex latex-pdf
breezedeus
breezedeus 开发者
3k
257
72
+10
排名 #4
1月28日
查看详情

rasbt/reasoning-from-scratch

Jupyter Notebook

使用PyTorch从零开始逐步实现一个推理大语言模型

ai artificial-intelligence deep-learning
rasbt
rasbt 开发者
2.8k
398
83
+29
排名 #11
2月3日
查看详情

opengeos/geoai

Python

GeoAI：面向地理空间数据的人工智能

ai data-science deep-learning
opengeos
opengeos 开发者
2.6k
361
39
+28
排名 #6
2月22日
查看详情

EnzymeAD/Enzyme

LLVM

面向LLVM与MLIR的高性能自动微分框架。

ad automatic-differentiation c
EnzymeAD
EnzymeAD 开发者
1.5k
152
85
+13
排名 #3
2月7日
查看详情

dvgodoy/PyTorchStepByStep

Jupyter Notebook

Official repository of my book: "Deep Learning with PyTorch Step-by-Step: A Beginner's Guide"

cnn-pytorch deep-learning python
dvgodoy
dvgodoy 开发者
1.4k
492
60
+12
排名 #14
2月15日
查看详情

SemiAnalysisAI/InferenceX

Python

开源持续推理基准测试：Qwen3.5、DeepSeek、GPTOSS - GB200 NVL72对比MI355X、B200、GB300 NVL72、H100，即将支持TPUv6e/v7/Trainium2/3

ai amd benchmark
SemiAnalysisAI
SemiAnalysisAI 开发者
500
86
79
+6
排名 #6
2月18日
查看详情

NVIDIA-NeMo/Automodel

Python

专为大语言模型/视觉语言模型设计的PyTorch分布式原生训练库，开箱即用支持Hugging Face

deepseek-v3-2 finetuning gemma3
NVIDIA-NeMo
NVIDIA-NeMo 开发者
321
70
76
+4
排名 #5
2月18日
查看详情
助手