L2 & L1 Regularizers in Trax

[ghost]

In Keras we can easily implement our regularizers such as : tf.keras.layers.Dense(3, kernel_regularizer='l2'). But I could not find such an attribute or function to do a similar task in Trax. It should be such an option in Trax, but maybe there are some other parameters for this task.

Thanks in advance.