Re-exports§
pub use config::DatasetConfig;pub use config::GenerationConfig;pub use config::LearningRateScheduleConfig;pub use config::ModelOverrides;pub use config::OptimizerConfig;pub use config::TrainingConfig;pub use config::TrainingHyperparameters;pub use config::load_training_config;pub use dataset::ShakespeareBatch;pub use dataset::ShakespeareDataset;pub use dataset::ShakespeareRandomDataLoader;pub use dataset::ShakespeareSplit;pub use kernel::BlockPattern1d;pub use kernel::BlockPattern2d;pub use kernel::BlockSparseConfig;pub use model::BDH;pub use model::BDHConfig;pub use model::language_model_loss;