Crate bunsen 

bunsen

bunsen aims to be a “batteries included” complementary community standard library for extending the burn tensor library.

Organization

Burn Extensions

• burner - this is a library of burn::module::Module lifecycle components that extend the current functionality of burn.
  • module::reflection has powerful tools for dynamic burn::module::Module reflection.
  • optim has parameter-group optimizer extensions.
• contracts - this is a library of runtime tensor-shape contracts.

Component Libraries

• blocks - this is a library of burn::module::Module components. This includes simple inner layers, recurrent utility blocks, and entire model families.
• ops - this is a library burn::tensor::Tensor operations.
• kits - this is a library of full models and simulation kits.

App and Testing Support Libs

• errors - this is a library of error types and tooling.
• support - this is a library of support functions for bunsen, including testing tooling which may be useful for clients.
• zspace - this is a library of z-space / index utilities.

Crate Features

• reflection (enabled by default) — Enable the XmlModuleTree reflection api for querying and transforming modules.
• store (enabled by default) — Enable the burn-store.
• train (enabled by default) — Enable features to use with burn/train.
• cache (enabled by default) — Enable the disk cache.
• testing (enabled by default) — Enables testing support features.
• cuda — Enable the cuda backend in the bunsen build. This will impact the bunsen::support::testing::PerformanceBackend choice. Dependents which enable backends for burn without enabling them for bunsen can still use those backends with bunsen; only bunse specific features will be disabled.
• metal — Enable the cuda backend in the bunsen build. This will impact the bunsen::support::testing::PerformanceBackend choice. Dependents which enable backends for burn without enabling them for bunsen can still use those backends with bunsen; only bunse specific features will be disabled.
• wgpu — Enable the cuda backend in the bunsen build. This will impact the bunsen::support::testing::PerformanceBackend choice. Dependents which enable backends for burn without enabling them for bunsen can still use those backends with bunsen; only bunse specific features will be disabled.
• flex (enabled by default) — Enable the cuda backend in the bunsen build. This will impact the bunsen::support::testing::PerformanceBackend choice. Dependents which enable backends for burn without enabling them for bunsen can still use those backends with bunsen; only bunse specific features will be disabled.

Modules

blocks

Neural-Network / Module Building Blocks

burner

burn-Adjacent Infrastructure

contracts

Shape Contracts

data

Management for Weight and Data IO.

errors

Common bunsen errors.

kits

Domain-Purpose Complete Kits

ops

Functional Tensor API

public

Re-exports public dependencies.

support

Bunsen / Client Support Utilities

zspace

Z-Space Index and Shape Support.