get_platform

static Config.get_platform()

Get the current platform of jax.

Return type:

str

Returns:

Current JAX platform, typically 'cpu' or 'gpu'.