style: cargo fmt --workspace (whitespace/wrapping only, no semantic change)
Whole-workspace rustfmt pass picked up while iterating on Mamba GPU backward work. Verified formatting-only via diff sampling; no logic changed. Co-Authored-By: Claude Sonnet 5 <[email protected]>
This commit is contained in:
@@ -14,19 +14,19 @@ pub use rtx_autograd as autograd;
|
||||
pub use rtx_tensor as tensor;
|
||||
|
||||
// Training
|
||||
pub use rtx_transformers as transformers;
|
||||
pub use rtx_compress as compress;
|
||||
pub use rtx_distributed as distributed;
|
||||
pub use rtx_flash_attention as flash_attention;
|
||||
pub use rtx_rl as rl;
|
||||
pub use rtx_transformers as transformers;
|
||||
|
||||
/// Training essentials
|
||||
pub mod prelude {
|
||||
pub use rtx_tensor::{Device, Tensor};
|
||||
// Use the correct exports from rtx_transformers
|
||||
pub use rtx_transformers::architectures::{TransformerArchitecture, TransformerConfig};
|
||||
pub use rtx_distributed::distributed_transformer_trainer::DistributedTransformerTrainer;
|
||||
pub use rtx_flash_attention::FlashAttention;
|
||||
pub use rtx_transformers::architectures::{TransformerArchitecture, TransformerConfig};
|
||||
// Use the correct exports from rtx_rl
|
||||
pub use rtx_rl::actor_learner::ActorLearner;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user