Crate burn_dragon_train 

Shared Dragon training/runtime helpers.

Preferred library-facing surface:

• api::config for backend/runtime/training config types
• api::runtime for train-time memory helpers
• api::wgpu for backend/device initialization helpers

Re-exports

pub use config::FsdpMixedPrecisionKind;

pub use config::GdpoConfig;

pub use config::GdpoHardGate;

pub use config::KernelSpec;

pub use config::LayerStateSpec;

pub use config::LowBitMemorySpec;

pub use config::LowBitModelSpec;

pub use config::LowBitSavedActivationInventorySpec;

pub use config::LowBitSavedActivationTensorSpec;

pub use config::ModelSpec;

pub use config::OptimizerSpec;

pub use config::ParallelCheckpointConfig;

pub use config::ParallelCheckpointFormat;

pub use config::ParallelCommunicationBackend;

pub use config::ParallelConfig;

pub use config::ParallelDataConfig;

pub use config::ParallelFsdpConfig;

pub use config::ParallelPipelineCacheConfig;

pub use config::ParallelPipelineConfig;

pub use config::ParallelSpec;

pub use config::ParallelTensorConfig;

pub use config::ParallelismKind;

pub use config::PipelineCacheEvictionKind;

pub use config::PipelineCachePolicy;

pub use config::PipelineCommunicationKind;

pub use config::PipelinePartitionKind;

pub use config::PipelineScheduleKind;

pub use config::PipelineSharedWeightSyncKind;

pub use config::PipelineTransportDtype;

pub use config::RunLayoutConfig;

pub use config::StateAxisSpec;

pub use config::StateLayout;

pub use config::StateTensorSpec;

pub use config::TensorParallelAxis;

pub use config::TensorParallelPartitionKind;

pub use config::VisionTeacherVariant;

pub use config::WgpuBackend;

pub use config::WgpuGenerationExecutor;

pub use config::WgpuInferenceConfig;

pub use config::WgpuMemoryConfig;

pub use config::WgpuRuntimeConfig;

pub use config::WgpuStartupAutotuneConfig;

pub use config::WgpuTrainingConfig;

Modules

api

Curated shared training/runtime API.

config

Shared training/runtime configuration helpers.

constants

Constants used across Dragon training/runtime adapters.

device

Device and backend resolution helpers.

wgpu

WGPU-specific runtime/device helpers.

Structs

SequenceKernelConfig