⚡️集成框架一览

将你最喜欢的框架与 SwanLab 结合使用！
下面是我们已集成的框架列表，欢迎提交 Issue 来反馈你想要集成的框架。

基础框架

• PyTorch
• MindSpore
• Keras

专有/微调框架

• PyTorch Lightning
• HuggingFace Transformers
• LLaMA Factory
• ModelScope Swift
• DiffSynth-Studio
• Sentence Transformers
• PaddleNLP
• OpenMind
• Torchtune
• XTuner
• MMEngine
• FastAI
• LightGBM
• XGBoost
• MLX-LM

评估框架

• EvalScope

计算机视觉

• Ultralytics
• MMDetection
• MMSegmentation
• PaddleDetection
• PaddleYOLO

强化学习

• Stable Baselines3
• veRL
• HuggingFace TRL
• EasyR1
• AReaL
• ROLL
• NVIDIA-NeMo RL
• RLinf
• MindSpeed-RL

其他框架

• TensorBoard
• Weights & Biases
• MLflow
• HuggingFace Accelerate
• Ray
• Hydra
• OmegaConf
• OpenAI
• ZhipuAI
• SpecForge