Crate burn_dragon_core 

Shared Dragon/Hatchling core math and state contracts.

Preferred library-facing surface:

• api::config
• api::state
• api::recurrent
• api::mhc
• api::experimental for research-only connector and sequence-kernel surfaces
• api::expert for lower-level implementation details

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::BDH;

pub use model::BDHConfig;

pub use model::BankedRhoState;

pub use model::BdhActivationThresholds;

pub use model::BdhBitNetDeployScaffold;

pub use model::BdhFiringTargetConfig;

pub use model::BdhFiringTargetKind;

pub use model::BdhInitializationConfig;

pub use model::BdhInitializationKind;

pub use model::BdhInitializer;

pub use model::BdhNeuronGainConfig;

pub use model::BdhNeuronGainKind;

pub use model::BdhProjectionRole;

pub use model::BdhResidualScalingConfig;

pub use model::BdhResidualScalingKind;

pub use model::BdhTopologyPriorConfig;

pub use model::BdhTopologyPriorKind;

pub use model::BitNetLowBitProtocol;

pub use model::BlockAttentionResidual;

pub use model::BlockAttentionResidualConfig;

pub use model::BlockAttentionResidualSummaryMode;

pub use model::ClockedSlowMemoryConfig;

pub use model::DragonNorm;

pub use model::DragonNormConfig;

pub use model::DragonNormKind;

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::LowBitActivationFormat;

pub use model::LowBitActivationGrouping;

pub use model::LowBitInferenceMode;

pub use model::LowBitKernelCapabilities;

pub use model::LowBitKernelFallbackReason;

pub use model::LowBitKernelPlan;

pub use model::LowBitKernelRuntimeKind;

pub use model::LowBitMemoryBucketEstimate;

pub use model::LowBitMemoryEstimateInput;

pub use model::LowBitNativeProjectionProfileSnapshot;

pub use model::LowBitProjectionPlan;

pub use model::LowBitQuantizationConfig;

pub use model::LowBitRhoConfig;

pub use model::LowBitSavedActivationCache;

pub use model::LowBitSavedActivationConfig;

pub use model::LowBitSavedActivationInventory;

pub use model::LowBitSavedActivationMode;

pub use model::LowBitSavedActivationRecomputePolicy;

pub use model::LowBitSavedActivationTensorInventoryEntry;

pub use model::LowBitTargetModule;

pub use model::LowBitTrainingMode;

pub use model::LowBitTrainingProjectionMemoryProfileSnapshot;

pub use model::LowBitTrainingProjectionMemoryStageSnapshot;

pub use model::LowBitTrainingQuantizeProfileSnapshot;

pub use model::LowBitWeightFormat;

pub use model::LowBitWeightGrouping;

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::PackedLowBitProjectionArtifacts;

pub use model::PackedRhoBlockEncoding;

pub use model::PackedRhoBlockState;

pub use model::PackedRhoInt8DeviceState;

pub use model::PackedSavedActivationBuffer;

pub use model::PackedSavedActivationState;

pub use model::ResidualConnectorKind;

pub use model::RhoCompressionConfig;

pub use model::RhoCompressionInterval;

pub use model::RhoCompressionQualityGate;

pub use model::RhoCompressionStatsSnapshot;

pub use model::RhoPrecisionConfig;

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::StructuredBankRole;

pub use model::StructuredDenseUpdateOutput;

pub use model::StructuredGridState;

pub use model::StructuredRouteOperation;

pub use model::StructuredRoutePattern;

pub use model::StructuredRouteSpec;

pub use model::StructuredRoutingSpec;

pub use model::StructuredStepMode;

pub use model::StructuredTopologyState;

pub use model::SummaryMemoryConfig;

pub use model::YNeuronRecurrenceConfig;

pub use model::build_low_bit_saved_activation_inventory;

pub use model::estimate_low_bit_memory_buckets;

pub use model::logits_projection_profile_reset;

pub use model::logits_projection_profile_snapshot;

pub use model::low_bit_kernel_capabilities;

pub use model::low_bit_kernel_capabilities_for_backend_name;

pub use model::low_bit_native_decoder_tail_profile_snapshot;

pub use model::low_bit_native_lowrank_profile_snapshot;

pub use model::low_bit_native_projection_profile_reset;

pub use model::low_bit_training_lowrank_memory_profile_snapshot;

pub use model::low_bit_training_quantize_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 model::pack_saved_activation_state;

pub use model::resolve_low_bit_kernel_plan;

pub use model::resolve_low_bit_kernel_plan_for_backend_name;

pub use model::rho_compression_profile_reset;

pub use model::rho_compression_profile_snapshot;

pub use model::rho_compression_snapshot_passes_gate;

pub use model::structured_dense_update_tokens;

pub use model::structured_predict_decay;

pub use model::target_major_apply_decay;

pub use model::target_major_decay_add;

pub use model::target_major_identity_read;

pub use model::target_major_identity_write;

pub use model::target_major_outer_product;

pub use model::unpack_saved_activation_state;

pub use positional::RotaryEmbedding;

Modules

api

Curated public surface for the Dragon core.

constants

experimental

kernel

model

positional

Enums

LowrankGradInputExecutor