wangzyon / trt_learn
TensorRT encapsulation, learn, rewrite, practice.
☆28Updated 2 years ago
Alternatives and similar repositories for trt_learn:
Users that are interested in trt_learn are comparing it to the libraries listed below
- async inference for machine learning model☆26Updated 2 years ago
- 使用 CUDA C++ 实现的 llama 模型推理框架☆54Updated 5 months ago
- ☆29Updated 5 months ago
- ☆24Updated last year
- EasyNN是一个面向教学而开发的神经网络推理框架,旨在让大家0基础也能自主完成推理框架编写!☆28Updated 8 months ago
- b站上的课程☆74Updated last year
- Quick and Self-Contained TensorRT Custom Plugin Implementation and Integration☆57Updated 10 months ago
- An onnx-based quantitation tool.☆71Updated last year
- 该代码与B站上的视频 https://www.bilibili.com/video/BV18L41197Uz/?spm_id_from=333.788&vd_source=eefa4b6e337f16d87d87c2c357db8ca7 相关联。☆67Updated last year
- Speed up image preprocess with cuda when handle image or tensorrt inference☆67Updated last month
- ☆36Updated 6 months ago
- Using pattern matcher in onnx model to match and replace subgraphs.☆79Updated last year
- ☆41Updated 10 months ago
- Llama3 Streaming Chat Sample☆22Updated last year
- ☆20Updated last month
- A light llama-like llm inference framework based on the triton kernel.☆112Updated last week
- ☢️ TensorRT 2023复赛——基于TensorRT-LLM的Llama模型推断加速优化☆47Updated last year
- ☆10Updated 9 months ago
- A bunch of coding tutorials for my Youtube videos on Neural Network Quantization.☆16Updated 11 months ago
- This is a repository to practice multi-thread programming in C++☆24Updated last year
- 彻底弄懂BP反向传播,15行代码,C++实现也简单,MNIST分类98.29%精度☆34Updated 3 years ago
- algorithm-cpp projects☆80Updated 2 years ago
- ☆11Updated 2 months ago
- TensorRT 2022复赛方案: 首个基于Transformer的图像重建模型MST++的TensorRT模型推断优化☆138Updated 2 years ago
- 使用 cutlass 实现 flash-attention 精简版,具有教学意义☆40Updated 8 months ago
- CUDA 6大并行计算模式 代码与笔记☆60Updated 4 years ago
- 一大波学习onnx的案例☆15Updated 7 months ago
- arm-neon☆90Updated 9 months ago
- c++实现的clip推理,模型有一点点改动,但是不大,改动和导出模型的代码可以在readme里找到,模型文件都在Releases里,包括AX650的模型。新增支持ChineseCLIP☆30Updated 4 months ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆64Updated 8 months ago