Index A | B | C | E | F | G | I | L | M | N | O | Q | R | S | T | U | V A AdagradMomentumOptimizerSpec (class in jax_tpu_embedding.sparsecore.lib.nn.embedding_spec) AdagradOptimizerSpec (class in jax_tpu_embedding.sparsecore.lib.nn.embedding_spec) AdamOptimizerSpec (class in jax_tpu_embedding.sparsecore.lib.nn.embedding_spec) auto_stack_tables() (in module jax_tpu_embedding.sparsecore.lib.nn.table_stacking) B b1 (LaPropOptimizerSpec attribute) b2 (LaPropOptimizerSpec attribute) beta (FTRLOptimizerSpec attribute) beta2 (AdagradMomentumOptimizerSpec attribute) beta_1 (AdamOptimizerSpec attribute) beta_2 (AdamOptimizerSpec attribute) C combiner (TableSpec attribute) E embedding_dim (TableSpec attribute) eps (LaPropOptimizerSpec attribute) epsilon (AdagradMomentumOptimizerSpec attribute) (AdamOptimizerSpec attribute) exponent (AdagradMomentumOptimizerSpec attribute) F FeatureSpec (class in jax_tpu_embedding.sparsecore.lib.nn.embedding_spec) FTRLOptimizerSpec (class in jax_tpu_embedding.sparsecore.lib.nn.embedding_spec) G get_decay_rate() (LaPropOptimizerSpec method) get_hyperparameters() (AdagradMomentumOptimizerSpec method) (AdamOptimizerSpec method) (FTRLOptimizerSpec method) (LaPropOptimizerSpec method) (OptimizerSpec method) get_learning_rate() (OptimizerSpec method) get_optimizer_primitive() (AdagradMomentumOptimizerSpec method) (AdagradOptimizerSpec method) (AdamOptimizerSpec method) (FTRLOptimizerSpec method) (LaPropOptimizerSpec method) (OptimizerSpec method) (SGDOptimizerSpec method) I id_transformation (FeatureSpec property) initial_accumulator_value (AdagradMomentumOptimizerSpec attribute) (AdagradOptimizerSpec attribute) (FTRLOptimizerSpec attribute) initial_linear_value (FTRLOptimizerSpec attribute) initial_momentum_value (AdagradMomentumOptimizerSpec attribute) initial_slot_value (LaPropOptimizerSpec attribute) initializer (TableSpec attribute) input_shape (FeatureSpec attribute) L l1_regularization_strength (FTRLOptimizerSpec attribute) l2_regularization_strength (FTRLOptimizerSpec attribute) LaPropOptimizerSpec (class in jax_tpu_embedding.sparsecore.lib.nn.embedding_spec) learning_rate (AdagradMomentumOptimizerSpec attribute) (AdagradOptimizerSpec attribute) (AdamOptimizerSpec attribute) (FTRLOptimizerSpec attribute) (LaPropOptimizerSpec attribute) (OptimizerSpec attribute) (SGDOptimizerSpec attribute) learning_rate_power (FTRLOptimizerSpec attribute) M max_ids_per_partition (TableSpec attribute) max_unique_ids_per_partition (TableSpec attribute) momentum (AdagradMomentumOptimizerSpec attribute) multiply_linear_by_learning_rate (FTRLOptimizerSpec attribute) N name (FeatureSpec attribute) (TableSpec attribute) O optimizer (TableSpec attribute) OptimizerSpec (class in jax_tpu_embedding.sparsecore.lib.nn.embedding_spec) output_shape (FeatureSpec attribute) Q quantization_config (TableSpec attribute) R rms_clip_threshold (LaPropOptimizerSpec attribute) S setting_in_stack (TableSpec property) SGDOptimizerSpec (class in jax_tpu_embedding.sparsecore.lib.nn.embedding_spec) short_name() (AdagradMomentumOptimizerSpec method) (AdagradOptimizerSpec method) (AdamOptimizerSpec method) (FTRLOptimizerSpec method) (LaPropOptimizerSpec method) (OptimizerSpec method) (SGDOptimizerSpec method) slot_variables_count() (OptimizerSpec method) slot_variables_initializers() (AdagradMomentumOptimizerSpec method) (AdagradOptimizerSpec method) (AdamOptimizerSpec method) (FTRLOptimizerSpec method) (LaPropOptimizerSpec method) (OptimizerSpec method) (SGDOptimizerSpec method) stack_tables() (in module jax_tpu_embedding.sparsecore.lib.nn.table_stacking) stacked_table_spec (TableSpec property) suggested_coo_buffer_size_per_device (TableSpec attribute) T table_spec (FeatureSpec attribute) TableSpec (class in jax_tpu_embedding.sparsecore.lib.nn.embedding_spec) U use_nesterov (AdagradMomentumOptimizerSpec attribute) V vocabulary_size (TableSpec attribute)