Skip to content

Port Encoder-Decoder Example from Jax AI Stack #107

@chapman20j

Description

@chapman20j

Port this tutorial

The tutorial provides guidance on defining and training an encoder-decoder model.

Additional features

  • Update the parameter definitions to be in a dataclass for consistency with other Bonsai models
  • Use KV-cache later in the tutorial for faster decoding

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions