#![cfg_attr(not(feature = "std"), no_std)]
#![warn(missing_docs)]
//! This library provides multiple tensor implementations hidden behind an easy to use API
//! that supports reverse mode automatic differentiation.
#[macro_use]
extern crate derive_new;
extern crate alloc;
mod tensor;
#[cfg(feature = "export_tests")]
#[allow(missing_docs)]
mod tests;
pub use half::{bf16, f16};
pub use tensor::*;
pub use burn_common::reader::Reader; // Useful so that backends don't have to add `burn_common` as
// a dependency so that they can implement the traits.
#[cfg(feature = "benchmark")]
/// This module provides benchmark utilities for easily and reliably run
/// benches on any function that is generic over a backend.
///
/// This can be useful to compare backends on inference or training speed
/// for your models.
pub mod benchmark;