⚡️Integrated Frameworks

Use your favorite framework with SwanLab! Below is a list of frameworks we have integrated, please submit Issue to feedback the framework you want to integrate.

Basic Frameworks

• PyTorch
• MindSpore
• Keras

Specialized/Fine-tuned Frameworks

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

Evaluation Frameworks

• EvalScope

Computer Vision

• Ultralytics
• MMDetection
• MMSegmentation
• PaddleDetection
• PaddleYOLO

Reinforcement Learning

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

Others:

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