is there any way to use kernel regularization in a flax conv

[BigBalloon8]

keras has:

tf.keras.layers.Conv2D(32, 3, kernel_regularizer=regularizers.l2(0.00004))

but flax does not give the option for a kernal regulizer do i have to implemnet from scratch or is there an altenative