Array library specific functions# jax_register_pytree_nodes() → None[source]# Register PyTree implementation for Array and Dimension. For more information see Working with JAX. Return type: None
