redocument

Author: t-kalinowski

NAMESPACE

@@ -292,6 +292,7 @@ export(layer_dot)
 export(layer_dropout)
 export(layer_einsum_dense)
 export(layer_embedding)
+export(layer_equalization)
 export(layer_feature_space)
 export(layer_flatten)
 export(layer_flax_module_wrapper)
@@ -323,16 +324,26 @@ export(layer_max_pooling_3d)
 export(layer_maximum)
 export(layer_mel_spectrogram)
 export(layer_minimum)
+export(layer_mix_up)
 export(layer_multi_head_attention)
 export(layer_multiply)
 export(layer_normalization)
 export(layer_permute)
 export(layer_pipeline)
+export(layer_rand_augment)
 export(layer_random_brightness)
+export(layer_random_color_degeneration)
+export(layer_random_color_jitter)
 export(layer_random_contrast)
 export(layer_random_crop)
 export(layer_random_flip)
+export(layer_random_grayscale)
+export(layer_random_hue)
+export(layer_random_posterization)
 export(layer_random_rotation)
+export(layer_random_saturation)
+export(layer_random_sharpness)
+export(layer_random_shear)
 export(layer_random_translation)
 export(layer_random_zoom)
 export(layer_repeat_vector)
@@ -508,6 +519,7 @@ export(op_custom_gradient)
 export(op_depthwise_conv)
 export(op_det)
 export(op_diag)
+export(op_diagflat)
 export(op_diagonal)
 export(op_diff)
 export(op_digitize)
@@ -698,6 +710,7 @@ export(op_tri)
 export(op_tril)
 export(op_triu)
 export(op_trunc)
+export(op_unravel_index)
 export(op_unstack)
 export(op_var)
 export(op_vdot)