scembedding_jax
