row-major matmul optimization

NanoGPT (124M) in 5 minutes

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.

A tutorial for CUDA&PyTorch

Depth Pro: Sharp Monocular Metric Depth in Less Than a Second.

CUDA Templates for Linear Algebra Subroutines

Helpful tools and examples for working with flex-attention

The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)

A high-throughput and memory-efficient inference and serving engine for LLMs

The official Meta Llama 3 GitHub site

Triton implementation of FlashAttention2 that adds Custom Masks.

Ring attention implementation with flash attention

NumPy aware dynamic Python compiler using LLVM

LLM101n: Let's build a Storyteller

InstantStyle: Free Lunch towards Style-Preserving in Text-to-Image Generation 🔥

LLM training in simple, raw C/CUDA

本实验的主要目的是基于遥感图像计算灰度共生矩阵，并基于该矩阵计算多种纹理特征。所有的计算结果已与ENVI结果进行对比，实验结果一致。

depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.

how to optimize some algorithm in cuda.

A latent text-to-image diffusion model

APISR: Anime Production Inspired Real-World Anime Super-Resolution (CVPR 2024)

[arXiv 2024] Follow-Your-Click: This repo is the official implementation of "Follow-Your-Click: Open-domain Regional Image Animation via Short Prompts"

PyQt5 implementation of YOLOv5 GUI

Add gui for YoloV5 using PyQt5

[CVPR 2024] Real-Time Open-Vocabulary Object Detection

图像质量评估

Code release for CVPR 2024 paper LEOD: Label-Efficient Object Detection for Event Cameras

A Pytorch implementation of StyleGAN2