Actions: jax-ml/jax
Actions
Showing runs from all workflows
23,648 workflow run results
23,648 workflow run results
max_version, dl_device, copy
kwargs in __dlpack__
to match Array API
CI
#49792:
Pull request #20198
synchronize
by
Micky774
jax.experimental.multihost_utils.global_array_to_host_local_array
CI
#49783:
Pull request #20707
opened
by
rajasekharporeddy
core.Token
a non-trivial class which wraps a jax.Array
. Currently, we use a singleton and empty core.token
object everywhere. After the change, tokens could be created and threaded in and out of computations to build up dependency.
CI
#49782:
Pull request #20706
synchronize
by
copybara-service
bot