Stars
[MICCAI 2023] DiffMIC: Dual-Guidance Diffusion Network for Medical Image Classification
PyTorch implementation of RCG https://arxiv.org/abs/2312.03701
Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation. (using diffusion for 3D medical image segmentation)
FlagScale is a large model toolkit based on open-sourced projects.
Release for Improved Denoising Diffusion Probabilistic Models
FlagAI (Fast LArge-scale General AI models) is a fast, easy-to-use and extensible toolkit for large-scale model.
Code for AAAI2021 paper: Few-Shot Learning for Multi-label Intent Detection.
Tutel MoE: Optimized Mixture-of-Experts Library, Support GptOss/DeepSeek/Kimi-K2/Qwen3 using FP8/NVFP4/MXFP4
Efficient Training (including pre-training and fine-tuning) for Big Models
Implementation for Joint Prediction of Meningioma Grade and Brain Invasion via Task-Aware Contrastive Learning
NestedFormer: Nested Modality-Aware Transformer for Brain Tumor Segmentation (MICCAI 2022)
Accessible large language models via k-bit quantization for PyTorch.
GLM-130B: An Open Bilingual Pre-Trained Model (ICLR 2023)
Baseline AI models for 3D csPCa detection/diagnosis in bpMRI
A tool for managing models from model.baa.ac.cn
bert_seq2seq的DDP版本，支持bert、roberta、nezha、t5、gpt2等模型，支持seq2seq、ner、关系抽取等任务，无需添加额外代码，轻松启动DDP多卡训练。
pytorch实现 Bert 做seq2seq任务，使用unilm方案,现在也可以做自动摘要，文本分类，情感分析，NER，词性标注等任务,支持t5模型，支持GPT2进行文章续写。
3D Medical Image Segmentation Models，集成各种医学图像分割模型的小框架，主要是3D，持续更新...
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Implementation of MlpMixer model, Original paper: MLP-Mixer: An all-MLP Architecture for Vision
Collect some papers about transformer with vision. Awesome Transformer with Computer Vision (CV)
This repo provides the official code for : 1) TransBTS: Multimodal Brain Tumor Segmentation Using Transformer (https://arxiv.org/abs/2103.04430) , accepted by MICCAI2021. 2) TransBTSV2: Towards Bet…