Hide navigation sidebar

Hide table of contents sidebar

Toggle site navigation sidebar

Flax-extra documentation

Toggle table of contents sidebar

Flax-extra documentation

Examples

PerceiverIO classification
PerceiverIO language modeling
PerceiverIO multimodal autoencoding

Tutorials

Model training and evaluation
Supervised and unsupervised learning
RNN LM with Flax and Redex combinators
Introduction to Redex
Introduction to Flax

Reference

Flax-extra
Toggle child pages in navigation
- flax_extra package
  Toggle child pages in navigation
  - flax_extra.checkpoint package
  - flax_extra.combinator package
  - flax_extra.data package
  - flax_extra.evaluation package
  - flax_extra.layer package
    Toggle child pages in navigation
    - flax_extra.layer.io module
  - flax_extra.model package
    Toggle child pages in navigation
    - flax_extra.model.perceiver package
    - flax_extra.model.rnn package
  - flax_extra.operator package
  - flax_extra.training package
  - flax_extra.batch module
  - flax_extra.console module
  - flax_extra.random module
  - flax_extra.util module
  - flax_extra.version module

Toggle table of contents sidebar

flax_extra.model.rnn package¶

flax_extra.operator package

flax_extra.model.perceiver package

Copyright © 2021, Andrei Nesterov | Built with Sphinx and @pradyunsg's Furo theme. | Show Source