keras-rs-nightly 0.2.2.dev202506210333__py3-none-any.whl → 0.2.2.dev202506240336__py3-none-any.whl

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 (6) hide show
  1. keras_rs/src/layers/embedding/jax/distributed_embedding.py +6 -4
  2. keras_rs/src/version.py +1 -1
  3. {keras_rs_nightly-0.2.2.dev202506210333.dist-info → keras_rs_nightly-0.2.2.dev202506240336.dist-info}/METADATA +1 -1
  4. {keras_rs_nightly-0.2.2.dev202506210333.dist-info → keras_rs_nightly-0.2.2.dev202506240336.dist-info}/RECORD +6 -6
  5. {keras_rs_nightly-0.2.2.dev202506210333.dist-info → keras_rs_nightly-0.2.2.dev202506240336.dist-info}/WHEEL +0 -0
  6. {keras_rs_nightly-0.2.2.dev202506210333.dist-info → keras_rs_nightly-0.2.2.dev202506240336.dist-info}/top_level.txt +0 -0
keras_rs/src/layers/embedding/jax/distributed_embedding.py CHANGED
@@ -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 # type: ignore
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/src/version.py CHANGED
@@ -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.dev202506210333"
4
+ __version__ = "0.2.2.dev202506240336"
5
5
 
6
6
 
7
7
  @keras_rs_export("keras_rs.version")
{keras_rs_nightly-0.2.2.dev202506210333.dist-info → keras_rs_nightly-0.2.2.dev202506240336.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: keras-rs-nightly
3
- Version: 0.2.2.dev202506210333
3
+ Version: 0.2.2.dev202506240336
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.dev202506210333.dist-info → keras_rs_nightly-0.2.2.dev202506240336.dist-info}/RECORD RENAMED
@@ -5,7 +5,7 @@ keras_rs/metrics/__init__.py,sha256=Qxpf6OFooIL9TIn2l3WgOea3HFRG0hq02glPAxtMZ9c,
5
5
  keras_rs/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
6
  keras_rs/src/api_export.py,sha256=RsmG-DvO-cdFeAF9W6LRzms0kvtm-Yp9BAA_d-952zI,510
7
7
  keras_rs/src/types.py,sha256=1A-oLRdX1-f2DsVZBcNl8qNsaH8pM-gnleLT9FWZWBw,1189
8
- keras_rs/src/version.py,sha256=DzvcXWfc27aWZFMQQJ_p3H6FeQ9JQru4PQYpirDiWhw,224
8
+ keras_rs/src/version.py,sha256=wwV5DvZRlqnvjck1PuVu5Am631ouA4BBtH3yoCEK1HQ,224
9
9
  keras_rs/src/layers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
10
  keras_rs/src/layers/embedding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
11
  keras_rs/src/layers/embedding/base_distributed_embedding.py,sha256=11GicbB6m0wsHJQXISp6lcUyACVVYFLFerluUJUjDFA,44265
@@ -15,7 +15,7 @@ keras_rs/src/layers/embedding/embed_reduce.py,sha256=c-MnEw1-KWs0jTf0JJ_ZBOY-9hR
15
15
  keras_rs/src/layers/embedding/jax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
16
16
  keras_rs/src/layers/embedding/jax/checkpoint_utils.py,sha256=wZ4I5WZVNg5WnrD2j7nhAXgLzDc7xMrUEkSAOx5Sz5c,3495
17
17
  keras_rs/src/layers/embedding/jax/config_conversion.py,sha256=Di1UzRwLgGHd7RuWYJMj2mCOr1u9MseFEWaYKnwD9Bs,16742
18
- keras_rs/src/layers/embedding/jax/distributed_embedding.py,sha256=urH6kfxuUnvOGy5x0CnqiruHyTDoXpgTJAquiz57a3E,35410
18
+ keras_rs/src/layers/embedding/jax/distributed_embedding.py,sha256=Ee8RqsjXKKQWLGrOHopup8y4b5kp8nhxd0OCoUC4jJE,35537
19
19
  keras_rs/src/layers/embedding/jax/embedding_lookup.py,sha256=8LigXjPr7uQaUOdZM6yoLGoPYdRcbkXkFeL_sJoQ6uQ,8223
20
20
  keras_rs/src/layers/embedding/jax/embedding_utils.py,sha256=EHrQjPLl94STLWf9g8Ew8nuwupXRq-a_QmvFlXV6G6A,20331
21
21
  keras_rs/src/layers/embedding/tensorflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -50,7 +50,7 @@ keras_rs/src/metrics/utils.py,sha256=fGTo8j0ykVE5Y3yQCS2orSFcHY20Uxt0NazyPsybUsw
50
50
  keras_rs/src/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
51
51
  keras_rs/src/utils/doc_string_utils.py,sha256=CmqomepmaYcvpACpXEXkrJb8DMnvIgmYK-lJ53lYarY,1675
52
52
  keras_rs/src/utils/keras_utils.py,sha256=dc-NFzs3a-qmRw0vBDiMslPLfrm9yymGduLWesXPhuY,2123
53
- keras_rs_nightly-0.2.2.dev202506210333.dist-info/METADATA,sha256=dNam-ywXjg6KiwO_QwyGUCbZrmhpx0sieGuCBTzNfs8,5273
54
- keras_rs_nightly-0.2.2.dev202506210333.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
55
- keras_rs_nightly-0.2.2.dev202506210333.dist-info/top_level.txt,sha256=pWs8X78Z0cn6lfcIb9VYOW5UeJ-TpoaO9dByzo7_FFo,9
56
- keras_rs_nightly-0.2.2.dev202506210333.dist-info/RECORD,,
53
+ keras_rs_nightly-0.2.2.dev202506240336.dist-info/METADATA,sha256=ByWqXJcLS8m9wwMyio15hKNIlpzDn113H60ezHd34IY,5273
54
+ keras_rs_nightly-0.2.2.dev202506240336.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
55
+ keras_rs_nightly-0.2.2.dev202506240336.dist-info/top_level.txt,sha256=pWs8X78Z0cn6lfcIb9VYOW5UeJ-TpoaO9dByzo7_FFo,9
56
+ keras_rs_nightly-0.2.2.dev202506240336.dist-info/RECORD,,