Skip to main content

Crate burn_dragon_core

Crate burn_dragon_core 

Source
Expand description

Shared Dragon core math and state contracts for the focused language-training repo.

Re-exports§

pub use model::LanguageModuleLrScaleTarget;
pub use kernel::BlockPattern1d;
pub use kernel::BlockPattern2d;
pub use kernel::BlockSparseConfig;
pub use model::AttentionResidual;
pub use model::AttentionResidualConfig;
pub use model::BlockAttentionResidual;
pub use model::BlockAttentionResidualConfig;
pub use model::BlockAttentionResidualSummaryMode;
pub use model::ClockedSlowMemoryConfig;
pub use model::DragonActivationThresholds;
pub use model::DragonConfig;
pub use model::DragonFiringTargetConfig;
pub use model::DragonFiringTargetKind;
pub use model::DragonInitializationConfig;
pub use model::DragonInitializationKind;
pub use model::DragonInitializer;
pub use model::DragonModel;
pub use model::DragonNeuronGainConfig;
pub use model::DragonNeuronGainKind;
pub use model::DragonNorm;
pub use model::DragonNormConfig;
pub use model::DragonNormKind;
pub use model::DragonProjectionRole;
pub use model::DragonReservoirInitializationConfig;
pub use model::DragonResidualScalingConfig;
pub use model::DragonResidualScalingKind;
pub use model::DragonTopologyPriorConfig;
pub use model::DragonTopologyPriorKind;
pub use model::FusedAttentionExecutor;
pub use model::FusedKernelConfig;
pub use model::FusedProjectionExecutor;
pub use model::HaltHead;
pub use model::LanguageHeadConfig;
pub use model::LanguageMhcLayerDiagnostics;
pub use model::LanguagePipelineState;
pub use model::LatentFanoutScheduleConfig;
pub use model::LogitsProjectionProfileSnapshot;
pub use model::LowRankResidualMemoryProfileSnapshot;
pub use model::LowRankResidualMemoryStageSnapshot;
pub use model::LowRankResidualOutput;
pub use model::LowRankResidualProfileSnapshot;
pub use model::MambaSequenceConfig;
pub use model::ManifoldHyperConnectionCoefficientPolicy;
pub use model::ManifoldHyperConnectionCoefficients;
pub use model::ManifoldHyperConnectionStreamCoefficients;
pub use model::ManifoldHyperConnectionStreamOutput;
pub use model::ManifoldHyperConnectionWidthOutput;
pub use model::ManifoldHyperConnections;
pub use model::ManifoldHyperConnectionsConfig;
pub use model::MicroTransformerBlock;
pub use model::ModelState;
pub use model::ResidualConnectorKind;
pub use model::SequenceKernelConfig;
pub use model::SequenceMemorySystem;
pub use model::SequenceTrainingExecutor;
pub use model::SharedLowrankActivationBatchStats;
pub use model::SharedLowrankContinualBackpropRuntime;
pub use model::SharedLowrankFeatureMetrics;
pub use model::SharedLowrankParamIds;
pub use model::SummaryMemoryConfig;
pub use model::YNeuronRecurrenceConfig;
pub use model::logits_projection_profile_reset;
pub use model::logits_projection_profile_snapshot;
pub use model::lowrank_residual_memory_profile_reset;
pub use model::lowrank_residual_memory_profile_snapshot;
pub use model::lowrank_residual_profile_reset;
pub use model::lowrank_residual_profile_snapshot;
pub use model::lowrank_residual_step;
pub use model::lowrank_residual_step_next;
pub use model::mhc_merge;
pub use model::mhc_merge_with_coefficients;
pub use model::mhc_passthrough;
pub use model::mhc_passthrough_with_coefficients;
pub use model::mhc_split;
pub use model::mhc_split_with_coefficients;
pub use model::near_critical_embedding_initializer;
pub use model::near_critical_embedding_std;
pub use model::near_critical_projection_std;
pub use model::near_critical_residual_output_std;
pub use positional::RotaryEmbedding;

Modules§

api
constants
experimental
Experimental core surfaces retained in the focused Dragon repo.
kernel
model
objective
positional

Enums§

LowrankGradInputExecutor