mod auto_tier;
mod flat;
mod from_bake;
mod from_dduf;
mod from_gguf;
mod from_mlx;
mod from_nemo;
mod from_pt;
mod manifest;
mod package;
mod placement;
mod seals;
mod store_zip;
mod tier;
mod verify;
mod weights_index;
mod write;
#[cfg(feature = "remote")]
mod remote;
pub use auto_tier::{AutoTierOptions, apply_auto_tier};
pub use flat::{
FLAG_BINCODE_TOC, FLAG_HYBRID, FLAT_CONTAINER_VERSION, FLAT_MAGIC, FlatHeader, FlatToc,
is_flat_magic,
};
pub use from_bake::{BakeWeight, package_from_bake};
pub use from_dduf::{DdufImportOptions, dduf_to_rlxp};
pub use from_gguf::{GgufImportOptions, gguf_to_rlxp};
pub use from_mlx::{MlxImportOptions, mlx_to_rlxp};
pub use from_nemo::{NemoImportOptions, nemo_to_rlxp};
pub use from_pt::{PtImportOptions, pt_to_rlxp};
pub use manifest::{
COMPAT_VERSION, DistRef, FORMAT_NAME, FORMAT_VERSION, GraphRef, Manifest, SidecarRef,
WeightsRef,
};
pub use package::{MaterializeMode, MemberSource, Package, dtype_for_weight_scheme};
pub use placement::{ExpertPlacement, Placement, TensorShard};
pub use seals::{SEAL_MAGIC, is_sealed, seal_bytes, unseal_bytes};
pub use tier::{
Codec, DEFAULT_WARM_BLOCK, StorageTier, checksum_hex, decode_zstd_block_at, decode_zstd_blocks,
decode_zstd_blocks_parallel, encode_zstd_blocks,
};
pub use verify::{VerifyReport, verify_package};
pub use weights_index::{WeightEntry, WeightsIndex};
pub use write::{
ContainerKind, PackedWeight, WriteOptions, graph_with_stripped_weights, infer_container,
infer_zip_from_path, write_package, write_package_dir, write_package_flat, write_package_zip,
};
#[cfg(feature = "remote")]
pub use remote::{RemoteFlat, http_range};