Skip to main content

Module network

burn_mamba::mamba1

Module network 

Source
Expand description

Utilizes Mamba1 and other Modules to build a Mamba1 model capable of utilizing the state-spaces/mamba-130m text prediction models.

References:

  • https://github.com/huggingface/candle/blob/fd7c8565646039e35925b8730d27ddad195d7e73/candle-examples/examples/mamba-minimal/
  • https://github.com/johnma2006/mamba-minimal/blob/61f01953ca153f8c4a850d7111beecbf4be9cee1/

Structsยง

Mamba1Network
Mamba1NetworkConfig
Mamba1NetworkRecord
The record type for the module.
Mamba1NetworkRecordItem
The record item type for the module.