File size: 201 Bytes
5ccb4fd | 1 2 3 4 5 6 7 8 9 10 11 12 | # Copyright (c) 2026 Simulacra Research Inc.
# SPDX-License-Identifier: Apache-2.0
import jax.numpy as jnp
def compute_dtype(*_args, **_kwargs):
return jnp.float32
__all__ = ["compute_dtype"]
|