keras-rs-nightly 0.2.2.dev202507250342__py3-none-any.whl → 0.2.2.dev202507260341__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 -1
  2. keras_rs/src/version.py +1 -1
  3. {keras_rs_nightly-0.2.2.dev202507250342.dist-info → keras_rs_nightly-0.2.2.dev202507260341.dist-info}/METADATA +1 -1
  4. {keras_rs_nightly-0.2.2.dev202507250342.dist-info → keras_rs_nightly-0.2.2.dev202507260341.dist-info}/RECORD +6 -6
  5. {keras_rs_nightly-0.2.2.dev202507250342.dist-info → keras_rs_nightly-0.2.2.dev202507260341.dist-info}/WHEEL +0 -0
  6. {keras_rs_nightly-0.2.2.dev202507250342.dist-info → keras_rs_nightly-0.2.2.dev202507260341.dist-info}/top_level.txt +0 -0
keras_rs/src/layers/embedding/jax/distributed_embedding.py CHANGED
@@ -221,9 +221,14 @@ class DistributedEmbedding(base_distributed_embedding.DistributedEmbedding):
221
221
  if jax.__version_info__ >= (0, 6, 3)
222
222
  else jax_layout.DeviceLocalLayout # type: ignore
223
223
  )
224
+ layout = (
225
+ LayoutClass(major_to_minor=(0, 1), tiling=((8,),)) # type: ignore
226
+ if jax.__version_info__ >= (0, 7, 1)
227
+ else LayoutClass(major_to_minor=(0, 1), _tiling=((8,),)) # type: ignore
228
+ )
224
229
  # pylint: disable-next=protected-access
225
230
  sparsecore_layout._backend_layout = jax_layout.Format(
226
- LayoutClass(major_to_minor=(0, 1), _tiling=((8,),)), # type: ignore
231
+ layout, # type: ignore
227
232
  jax.sharding.NamedSharding(
228
233
  device_mesh.backend_mesh,
229
234
  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.dev202507250342"
4
+ __version__ = "0.2.2.dev202507260341"
5
5
 
6
6
 
7
7
  @keras_rs_export("keras_rs.version")
{keras_rs_nightly-0.2.2.dev202507250342.dist-info → keras_rs_nightly-0.2.2.dev202507260341.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.dev202507250342
3
+ Version: 0.2.2.dev202507260341
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.dev202507250342.dist-info → keras_rs_nightly-0.2.2.dev202507260341.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=VZKgvPv9oFsKFzLKFql97hPVKh1knpk_A7ZiJ2Bx6hA,224
8
+ keras_rs/src/version.py,sha256=rKddhfjQGFSUzxrs3TRgwF3CAjISC7D-RQwzvhIOd5w,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=Hr54K8Y3OpUQc7ExtftLONoY7w4x4R08BkkMNtrpbZw,45026
@@ -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=Ee8RqsjXKKQWLGrOHopup8y4b5kp8nhxd0OCoUC4jJE,35537
18
+ keras_rs/src/layers/embedding/jax/distributed_embedding.py,sha256=jLqEuh_7hHM4bacyMY6FYpzlCUd9HcJHmnOG3ouucuQ,35734
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.dev202507250342.dist-info/METADATA,sha256=7XlCZBPApW-RKiBUDFJNFVemdE6LvWwyRtNSCCg19mI,5273
54
- keras_rs_nightly-0.2.2.dev202507250342.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
55
- keras_rs_nightly-0.2.2.dev202507250342.dist-info/top_level.txt,sha256=pWs8X78Z0cn6lfcIb9VYOW5UeJ-TpoaO9dByzo7_FFo,9
56
- keras_rs_nightly-0.2.2.dev202507250342.dist-info/RECORD,,
53
+ keras_rs_nightly-0.2.2.dev202507260341.dist-info/METADATA,sha256=Nkl9QgHgzwUHP5WvL_LJY_L3PzWYN25AB5LbMGsCiiw,5273
54
+ keras_rs_nightly-0.2.2.dev202507260341.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
55
+ keras_rs_nightly-0.2.2.dev202507260341.dist-info/top_level.txt,sha256=pWs8X78Z0cn6lfcIb9VYOW5UeJ-TpoaO9dByzo7_FFo,9
56
+ keras_rs_nightly-0.2.2.dev202507260341.dist-info/RECORD,,