Author
Label
Projects
Milestones
Reviews
Assignee
Sort
[jax2tf] Added testing for the conversion of convert_element_type.
cla: yes
#5083
opened Dec 2, 2020 by
bchetioui
Ensure lax.convert_element_type preserves concrete inputs
cla: yes
#5075
opened Dec 2, 2020 by
jakevdp
all_to_all: Add optional axis_index_groups argument
cla: yes
#5059
opened Dec 1, 2020 by
jpuigcerver
Expose sharded_jit, PartitionSpec, and with_sharding_constraint in jax.experimental namespace.
cla: yes
pull ready
#5038
opened Nov 28, 2020 by
skye
[jax2tf] Add note that the TF.js conversion is experimental.
cla: yes
#4997
opened Nov 23, 2020 by
bchetioui
Extract cloud TPU Colab set-up to a re-usable helper.
cla: yes
#4911
opened Nov 16, 2020 by
aslanides
ci-build.yaml: mypy jax failing on the current version of Python
cla: yes
#4897
opened Nov 15, 2020 by
cclauss
Previous Next
ProTip!
Follow long discussions with comments:>50.