`brainpy.jaxsetting` module

`enable_x64`(mode)
`set_platform`(platform)	Changes platform to CPU, GPU, or TPU.
`set_host_device_count`(n)	By default, XLA considers all CPU cores as one device.

brainpy.jaxsetting module