flax.linen

Imported 19 times

  1. 13 × from flax import linen as nn
  2. 6 × import flax.linen as nn