Replace jnp.ndarray with Tensor from axlearn.common.utils#973

Merged

markblee merged 1 commit intoapple:mainfrom apivovarov:Tensor

Feb 11, 2025

Commits on Feb 9, 2025

Replace jnp.ndarray with Tensor from axlearn.common.utils
apivovarov
committed