keras-rs-nightly 0.2.2.dev202508040349__tar.gz → 0.2.2.dev202508060343__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of keras-rs-nightly might be problematic. Click here for more details.

Files changed (61) hide show
  1. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/PKG-INFO +1 -1
  2. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/base_distributed_embedding.py +4 -4
  3. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/distributed_embedding_config.py +4 -1
  4. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/version.py +1 -1
  5. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs_nightly.egg-info/PKG-INFO +1 -1
  6. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/README.md +0 -0
  7. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/api/__init__.py +0 -0
  8. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/api/layers/__init__.py +0 -0
  9. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/api/losses/__init__.py +0 -0
  10. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/api/metrics/__init__.py +0 -0
  11. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/__init__.py +0 -0
  12. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/api_export.py +0 -0
  13. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/__init__.py +0 -0
  14. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/__init__.py +0 -0
  15. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/distributed_embedding.py +0 -0
  16. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/embed_reduce.py +0 -0
  17. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/jax/__init__.py +0 -0
  18. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/jax/checkpoint_utils.py +0 -0
  19. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/jax/config_conversion.py +0 -0
  20. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/jax/distributed_embedding.py +0 -0
  21. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/jax/embedding_lookup.py +0 -0
  22. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/jax/embedding_utils.py +0 -0
  23. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/tensorflow/__init__.py +0 -0
  24. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/tensorflow/config_conversion.py +0 -0
  25. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/tensorflow/distributed_embedding.py +0 -0
  26. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/feature_interaction/__init__.py +0 -0
  27. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/feature_interaction/dot_interaction.py +0 -0
  28. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/feature_interaction/feature_cross.py +0 -0
  29. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/retrieval/__init__.py +0 -0
  30. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/retrieval/brute_force_retrieval.py +0 -0
  31. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/retrieval/hard_negative_mining.py +0 -0
  32. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/retrieval/remove_accidental_hits.py +0 -0
  33. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/retrieval/retrieval.py +0 -0
  34. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/retrieval/sampling_probability_correction.py +0 -0
  35. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/losses/__init__.py +0 -0
  36. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/losses/pairwise_hinge_loss.py +0 -0
  37. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/losses/pairwise_logistic_loss.py +0 -0
  38. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/losses/pairwise_loss.py +0 -0
  39. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/losses/pairwise_loss_utils.py +0 -0
  40. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/losses/pairwise_mean_squared_error.py +0 -0
  41. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/losses/pairwise_soft_zero_one_loss.py +0 -0
  42. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/__init__.py +0 -0
  43. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/dcg.py +0 -0
  44. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/mean_average_precision.py +0 -0
  45. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/mean_reciprocal_rank.py +0 -0
  46. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/ndcg.py +0 -0
  47. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/precision_at_k.py +0 -0
  48. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/ranking_metric.py +0 -0
  49. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/ranking_metrics_utils.py +0 -0
  50. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/recall_at_k.py +0 -0
  51. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/metrics/utils.py +0 -0
  52. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/types.py +0 -0
  53. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/utils/__init__.py +0 -0
  54. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/utils/doc_string_utils.py +0 -0
  55. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/utils/keras_utils.py +0 -0
  56. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs_nightly.egg-info/SOURCES.txt +0 -0
  57. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs_nightly.egg-info/dependency_links.txt +0 -0
  58. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs_nightly.egg-info/requires.txt +0 -0
  59. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs_nightly.egg-info/top_level.txt +0 -0
  60. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/pyproject.toml +0 -0
  61. {keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/setup.cfg +0 -0
{keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: keras-rs-nightly
3
- Version: 0.2.2.dev202508040349
3
+ Version: 0.2.2.dev202508060343
4
4
  Summary: Multi-backend recommender systems with Keras 3.
5
5
  Author-email: Keras team <keras-users@googlegroups.com>
6
6
  License: Apache License 2.0
{keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/base_distributed_embedding.py RENAMED
@@ -146,14 +146,14 @@ class DistributedEmbedding(keras.layers.Layer):
146
146
  feature1 = keras_rs.layers.FeatureConfig(
147
147
  name="feature1",
148
148
  table=table1,
149
- input_shape=(PER_REPLICA_BATCH_SIZE,),
150
- output_shape=(PER_REPLICA_BATCH_SIZE, TABLE1_EMBEDDING_SIZE),
149
+ input_shape=(GLOBAL_BATCH_SIZE,),
150
+ output_shape=(GLOBAL_BATCH_SIZE, TABLE1_EMBEDDING_SIZE),
151
151
  )
152
152
  feature2 = keras_rs.layers.FeatureConfig(
153
153
  name="feature2",
154
154
  table=table2,
155
- input_shape=(PER_REPLICA_BATCH_SIZE,),
156
- output_shape=(PER_REPLICA_BATCH_SIZE, TABLE2_EMBEDDING_SIZE),
155
+ input_shape=(GLOBAL_BATCH_SIZE,),
156
+ output_shape=(GLOBAL_BATCH_SIZE, TABLE2_EMBEDDING_SIZE),
157
157
  )
158
158
 
159
159
  feature_configs = {
{keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/layers/embedding/distributed_embedding_config.py RENAMED
@@ -102,7 +102,10 @@ class FeatureConfig:
102
102
  input_shape: The input shape of the feature. The feature fed into the
103
103
  layer has to match the shape. Note that for ragged dimensions in the
104
104
  input, the dimension provided here presents the maximum value;
105
- anything larger will be truncated.
105
+ anything larger will be truncated. Also note that the first
106
+ dimension represents the global batch size. For example, on TPU,
107
+ this represents the total number of samples that are dispatched to
108
+ all the TPUs connected to the current host.
106
109
  output_shape: The output shape of the feature activation. What is
107
110
  returned by the embedding layer has to match this shape.
108
111
  """
{keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs/src/version.py RENAMED
@@ -1,7 +1,7 @@
1
1
  from keras_rs.src.api_export import keras_rs_export
2
2
 
3
3
  # Unique source of truth for the version number.
4
- __version__ = "0.2.2.dev202508040349"
4
+ __version__ = "0.2.2.dev202508060343"
5
5
 
6
6
 
7
7
  @keras_rs_export("keras_rs.version")
{keras_rs_nightly-0.2.2.dev202508040349 → keras_rs_nightly-0.2.2.dev202508060343}/keras_rs_nightly.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: keras-rs-nightly
3
- Version: 0.2.2.dev202508040349
3
+ Version: 0.2.2.dev202508060343
4
4
  Summary: Multi-backend recommender systems with Keras 3.
5
5
  Author-email: Keras team <keras-users@googlegroups.com>
6
6
  License: Apache License 2.0