Pull requests: jax-ml/jax
Pull requests list
jax.experimental.tree_math
cla: yes
pull ready
Ready for copybara import and testing
#8504
by shoyer
was closed Dec 29, 2021
Loading…
add core.cur_level() function to get trace level
pull ready
Ready for copybara import and testing
#9423
by mattjj
was closed Jan 5, 2023
Loading…
1 task
[shard-map] first draft of shard_map tutorial
pull ready
Ready for copybara import and testing
#19280
by mattjj
was merged Jan 18, 2024
Loading…
leak checker: explain why a tracer is alive by printing a reference chain
pull ready
Ready for copybara import and testing
#13022
by mattjj
was merged Oct 28, 2022
Loading…
Add configuration options jax_cuda_visible_devices and jax_rocm_visib…
pull ready
Ready for copybara import and testing
#11672
by hawkinsp
was merged Aug 8, 2022
Loading…
while_loop vmap init_val with mixed map/unmap dims
cla: yes
#495
by mattjj
was merged Mar 11, 2019
Loading…
Add constant handler for tokens.
pull ready
Ready for copybara import and testing
#9559
by hawkinsp
was merged Feb 14, 2022
Loading…
ProTip!
Filter pull requests by the default branch with base:main.