11rocky

Pinned Repositories

tensorrt_yolo
yolov3/v4 on tensorrt, with fp32, fp16, int8 support, dynamic batch, image pre-process on gpu
Language:C++41
deep_learning
a deep learning framework with pytorch DDP
Language:Python00
flash_attention
implemention of flash attention v1 and v2 with numpy
Language:Python00
magic_enum
Static reflection for enums (to string, from string, iteration) for modern C++, work with any enum type without any macro or boilerplate code
Language:C++00
zhangyachen.github.io
zhangyachen's blog
Language:JavaScript00

11rocky's Repositories

11rocky/flash_attention
implemention of flash attention v1 and v2 with numpy
Language:Python
11rocky/deep_learning
a deep learning framework with pytorch DDP
Language:Python
11rocky/tensorrt_yolo
yolov3/v4 on tensorrt, with fp32, fp16, int8 support, dynamic batch, image pre-process on gpu
Language:C++41
11rocky/magic_enum
Static reflection for enums (to string, from string, iteration) for modern C++, work with any enum type without any macro or boilerplate code
11rocky/zhangyachen.github.io
zhangyachen's blog