Index

A

apply() (blox.RecurrenceBase method) (blox.SequenceBase method)

attend() (blox.Embed method) avg_pool() (in module blox)

B

BatchNorm (class in blox) blox module

blox.blocks module blox.visualize module

C

cell (blox.LSTMState attribute) child() (blox.Graph method)

Conv (class in blox) ConvTranspose (class in blox)

D

display() (in module blox)

Dropout (class in blox) dynamic_scan() (in module blox)

E

Embed (class in blox)

G

get_counter() (blox.Rng method) get_param() (blox.Module method) get_seed() (blox.Rng method)

Graph (class in blox) GRU (class in blox) GRUState (class in blox)

H

hidden (blox.GRUState attribute) (blox.LSTMState attribute)

I

initial_state() (blox.GRU method) (blox.LSTM method) (blox.SequenceBase method)

is_locked (blox.Params property) is_static (blox.RecurrenceBase property) items() (blox.Params method)

K

keys() (blox.Params method)

L

LayerNorm (class in blox) Linear (class in blox)

locked() (blox.Params method) LSTM (class in blox) LSTMState (class in blox)

M

max_pool() (in module blox) maybe_reset_state() (blox.RecurrenceBase method) merge() (blox.Params method) metadata (blox.Param attribute) min_pool() (in module blox)

module blox blox.blocks blox.visualize module (blox.Graph property) Module (class in blox)

P

Param (class in blox) param_path() (blox.Module method)

Params (class in blox) parent (blox.Graph property)

R

RecurrenceBase (class in blox) replace() (blox.Param method)

RMSNorm (class in blox) Rng (class in blox) root (blox.Graph property)

S

seed() (blox.Rng method) SequenceBase (class in blox) Sequential (class in blox)

set_param() (blox.Module method) sharding (blox.Param property) split() (blox.Params method) static_scan() (in module blox)

T

trainable (blox.Param attribute) tree_flatten() (blox.Param method) (blox.Params method)

tree_unflatten() (blox.Param class method) (blox.Params class method)

U

unlocked() (blox.Params method)

V

value (blox.Param attribute)

values() (blox.Params method)

W

walk() (blox.Graph method)