
	f
	`flax_extra`
	`flax_extra.batch`
	`flax_extra.checkpoint`
	`flax_extra.combinator`
	`flax_extra.console`
	`flax_extra.data`
	`flax_extra.evaluation`
	`flax_extra.layer`
	`flax_extra.layer.io`
	`flax_extra.model`
	`flax_extra.model.perceiver`
	`flax_extra.model.rnn`
	`flax_extra.operator`
	`flax_extra.random`
	`flax_extra.training`
	`flax_extra.util`
	`flax_extra.version`

Python Module Index