Pinned Repositories
graph-learn
An Industrial Graph Neural Network Framework
cub
Cooperative primitives for CUDA C++.
DiT
Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"
FlashModels
Fast and easy distributed model training examples.
graph-learn
graph-learn
graphlearn-for-pytorch
A graph learning library for PyTorch that makes distributed GNN training and inference easy and efficient.
openxla-xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
ms-swift
Use PEFT or Full-parameter to finetune 300+ LLMs or 80+ MLLMs. (Qwen2, GLM4v, Internlm2.5, Yi, Llama3.1, Llava-Video, Internvl2, MiniCPM-V-2.6, Deepseek, Baichuan2, Gemma2, Phi3-Vision, ...)
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
baoleai's Repositories
baoleai/cub
Cooperative primitives for CUDA C++.
baoleai/DiT
Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"
baoleai/FlashModels
Fast and easy distributed model training examples.
baoleai/graph-learn
graph-learn
baoleai/graphlearn-for-pytorch
A graph learning library for PyTorch that makes distributed GNN training and inference easy and efficient.
baoleai/openxla-xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
baoleai/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
baoleai/pytorch_geometric
Graph Neural Network Library for PyTorch
baoleai/xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
baoleai/swift
魔搭大模型训练推理工具箱，支持LLaMA、千问、ChatGLM、BaiChuan等多种模型及LoRA等多种训练方式(The LLM training/inference framework of ModelScope community, Support various models like LLaMA, Qwen, ChatGLM, Baichuan and others, and training methods like LoRA, ResTuning, NEFTune, etc.)
baoleai/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.