keras-rs-nightly 0.2.2.dev202506200334__tar.gz → 0.2.2.dev202506230341__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.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/PKG-INFO +1 -1
  2. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/jax/distributed_embedding.py +6 -4
  3. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/version.py +1 -1
  4. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs_nightly.egg-info/PKG-INFO +1 -1
  5. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/README.md +0 -0
  6. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/api/__init__.py +0 -0
  7. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/api/layers/__init__.py +0 -0
  8. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/api/losses/__init__.py +0 -0
  9. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/api/metrics/__init__.py +0 -0
  10. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/__init__.py +0 -0
  11. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/api_export.py +0 -0
  12. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/__init__.py +0 -0
  13. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/__init__.py +0 -0
  14. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/base_distributed_embedding.py +0 -0
  15. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/distributed_embedding.py +0 -0
  16. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/distributed_embedding_config.py +0 -0
  17. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/embed_reduce.py +0 -0
  18. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/jax/__init__.py +0 -0
  19. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/jax/checkpoint_utils.py +0 -0
  20. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/jax/config_conversion.py +0 -0
  21. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/jax/embedding_lookup.py +0 -0
  22. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/jax/embedding_utils.py +0 -0
  23. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/tensorflow/__init__.py +0 -0
  24. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/tensorflow/config_conversion.py +0 -0
  25. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/tensorflow/distributed_embedding.py +0 -0
  26. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/feature_interaction/__init__.py +0 -0
  27. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/feature_interaction/dot_interaction.py +0 -0
  28. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/feature_interaction/feature_cross.py +0 -0
  29. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/retrieval/__init__.py +0 -0
  30. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/retrieval/brute_force_retrieval.py +0 -0
  31. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/retrieval/hard_negative_mining.py +0 -0
  32. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/retrieval/remove_accidental_hits.py +0 -0
  33. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/retrieval/retrieval.py +0 -0
  34. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/retrieval/sampling_probability_correction.py +0 -0
  35. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/losses/__init__.py +0 -0
  36. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/losses/pairwise_hinge_loss.py +0 -0
  37. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/losses/pairwise_logistic_loss.py +0 -0
  38. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/losses/pairwise_loss.py +0 -0
  39. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/losses/pairwise_loss_utils.py +0 -0
  40. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/losses/pairwise_mean_squared_error.py +0 -0
  41. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/losses/pairwise_soft_zero_one_loss.py +0 -0
  42. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/__init__.py +0 -0
  43. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/dcg.py +0 -0
  44. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/mean_average_precision.py +0 -0
  45. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/mean_reciprocal_rank.py +0 -0
  46. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/ndcg.py +0 -0
  47. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/precision_at_k.py +0 -0
  48. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/ranking_metric.py +0 -0
  49. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/ranking_metrics_utils.py +0 -0
  50. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/recall_at_k.py +0 -0
  51. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/metrics/utils.py +0 -0
  52. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/types.py +0 -0
  53. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/utils/__init__.py +0 -0
  54. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/utils/doc_string_utils.py +0 -0
  55. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/utils/keras_utils.py +0 -0
  56. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs_nightly.egg-info/SOURCES.txt +0 -0
  57. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs_nightly.egg-info/dependency_links.txt +0 -0
  58. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs_nightly.egg-info/requires.txt +0 -0
  59. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs_nightly.egg-info/top_level.txt +0 -0
  60. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/pyproject.toml +0 -0
  61. {keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/setup.cfg +0 -0
{keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: keras-rs-nightly
3
- Version: 0.2.2.dev202506200334
3
+ Version: 0.2.2.dev202506230341
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.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/keras_rs/src/layers/embedding/jax/distributed_embedding.py RENAMED
@@ -216,12 +216,14 @@ class DistributedEmbedding(base_distributed_embedding.DistributedEmbedding):
216
216
  )
217
217
  sparsecore_layout = keras.distribution.TensorLayout(axes, device_mesh)
218
218
  # Custom sparsecore layout with tiling.
219
+ LayoutClass = (
220
+ jax_layout.Layout
221
+ if jax.__version_info__ >= (0, 6, 3)
222
+ else jax_layout.DeviceLocalLayout
223
+ )
219
224
  # pylint: disable-next=protected-access
220
225
  sparsecore_layout._backend_layout = jax_layout.Format(
221
- jax_layout.DeviceLocalLayout(
222
- major_to_minor=(0, 1),
223
- _tiling=((8,),),
224
- ),
226
+ LayoutClass(major_to_minor=(0, 1), _tiling=((8,),)), # type: ignore
225
227
  jax.sharding.NamedSharding(
226
228
  device_mesh.backend_mesh,
227
229
  jax.sharding.PartitionSpec(
{keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/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.dev202506200334"
4
+ __version__ = "0.2.2.dev202506230341"
5
5
 
6
6
 
7
7
  @keras_rs_export("keras_rs.version")
{keras_rs_nightly-0.2.2.dev202506200334 → keras_rs_nightly-0.2.2.dev202506230341}/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.dev202506200334
3
+ Version: 0.2.2.dev202506230341
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