Skip to content

Pull requests: google/jax

New pull request New
206 Open 6,540 Closed
206 Open 6,540 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[x64] make host_callback_test compatible with strict dtype promotion
#11135 opened Jun 16, 2022 by jakevdp Loading…
@jakevdp
[x64] make jaxpr_effects_test pass with strict dtype promotion
pull ready
#11134 opened Jun 16, 2022 by jakevdp Loading…
@jakevdp
[x64] make lax_numpy_vectorize_test work with strict dtype promotion
pull ready
#11132 opened Jun 16, 2022 by jakevdp Loading…
@jakevdp
[x64] make lax_numpy_indexing_test pass with strict dtype promotion
kokoro:force-run pull ready
#11131 opened Jun 16, 2022 by jakevdp Loading…
@jakevdp
[x64] make checkify compatible with strict dtype promotion
#11129 opened Jun 16, 2022 by jakevdp Loading…
@jakevdp
[x64] make nn_test pass with strict dtype promotion
pull ready
#11126 opened Jun 16, 2022 by jakevdp Loading…
@jakevdp
[x64] make linalg functions & tests compatible with strict dtype promotion
pull ready
#11124 opened Jun 16, 2022 by jakevdp Loading…
@jakevdp
1
Revert unit test enablement for ROCm (from 8d9f17df19a647577357398c7a…
#11123 opened Jun 16, 2022 by rsanthanam-amd Loading…
Take aval instead of shape in __init__ of Array. This is because avals can have weak_types too that currently is not supported since we take Shape. Also add __repr__ to Array to make it easier to print Arrays. __repr__ works exactly as it does for DA and SDA when it is fully addressable. Otherwise it works like GDA.
#11121 opened Jun 16, 2022 by copybara-service bot Loading…
[dynamic-shapes] initial support for dynamic shape typechecks
pull ready
#11116 opened Jun 15, 2022 by mattjj Loading…
13
Enable bitwise reductions for integer dtypes in lax.reduce
#11115 opened Jun 15, 2022 by nicholasjng Loading…
1
Add tree_labels, tree_print.
#11111 opened Jun 15, 2022 by awf Loading…
5
First CL to integrate jax.Array into pmap.
#11106 opened Jun 15, 2022 by copybara-service bot Loading…
[x64] make scipy_ndimage_test compatible with strict dtype promotion
#11104 opened Jun 14, 2022 by jakevdp Loading…
@jakevdp
2
Implement Chebyshev class and functions
#11093 opened Jun 14, 2022 by gderossi Loading…
34
Fix zoom state for next iteration in line search
#11090 opened Jun 14, 2022 by david-anton Loading…
1
[mhlo] Unifiy different versions of round op (viz., mhlo::round_nearest_afz & mhlo::round_nearest_even) into one.
#11086 opened Jun 14, 2022 by copybara-service bot Loading…
Add max step size argument to odeint
#11077 opened Jun 13, 2022 by DanPuzzuoli Loading…
1
snapshot the threefry PRNG implementation
pull ready
#11061 opened Jun 10, 2022 by froystig • Draft
@froystig
Extend Common-Gotchas docs: speed of jnp.array
#11054 opened Jun 10, 2022 by onnoeberhard Loading…
1
3
Faster uniform floats on TPUs:
#11034 opened Jun 9, 2022 by copybara-service bot Loading…
Add jax.logging module
#10968 opened Jun 3, 2022 by nicholasjng Loading…
1
1
Add initial LOBPCG top-k eigenvalue solver (#3112)
#10962 opened Jun 3, 2022 by vlad17 Loading…
14
add scipy.special.factorial and scipy.special.comb
pull ready
#10948 opened Jun 2, 2022 by froystig • Draft
@froystig
3
[mhlo] create builder with type infer regardless of region.
#10940 opened Jun 1, 2022 by copybara-service bot Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.