Skip to content

Pull requests: google/jax

New pull request New
270 Open 9,071 Closed
270 Open 9,071 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 Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[NVIDIA GPU] Add missing file for the nightly CI.
#15167 opened Mar 23, 2023 by nouiz Loading…
[jax2tf] Minor addition to the documentation
#15166 opened Mar 23, 2023 by copybara-service bot Loading…
@gnecula
[jax2tf] Create a jax_export library with JAX-only pieces for native serialization
#15165 opened Mar 23, 2023 by copybara-service bot Loading…
@gnecula
[jax2tf] Turn an error into a warning with native serialization
#15160 opened Mar 23, 2023 by copybara-service bot Loading…
@gnecula
[jax2tf] A simple failing test on TPU with native serialization
#15159 opened Mar 23, 2023 by copybara-service bot Loading…
@gnecula
Migrate random_gamma_grad off xla_fallback
#15158 opened Mar 23, 2023 by copybara-service bot Loading…
@hawkinsp
Migrate igamma_grad_a_p off xla_fallback
#15157 opened Mar 23, 2023 by copybara-service bot Loading…
@hawkinsp
Refactor special functions into their own module.
#15153 opened Mar 22, 2023 by copybara-service bot Loading…
Migrate igamma_p off xla_fallback
#15152 opened Mar 22, 2023 by copybara-service bot Loading…
@hawkinsp
Delete xla_call_p since it has been replaced with pjit.pjit_p
#15150 opened Mar 22, 2023 by copybara-service bot Loading…
Add initial implementation of a run_state primitive
#15149 opened Mar 22, 2023 by sharadmv Loading…
2
Move jax._src.typing into a separate Bazel target.
#15147 opened Mar 22, 2023 by copybara-service bot Loading…
@hawkinsp
Migrate igammac_p off xla_fallback path
#15142 opened Mar 22, 2023 by copybara-service bot Loading…
@hawkinsp
Convolution functions in TF, like- tf.nn.depthwise_conv2d_v2, tf.nn.conv2d_transpose_v2, tf.nn.conv2d_v2 all follow the same principal when it comes to padding(explained here- https://www.tensorflow.org/api_docs/python/tf/nn#notes_on_padding_2). These principal happens to match with that of lax.convolution.conv_general_dilated.
#15130 opened Mar 21, 2023 by copybara-service bot Loading…
1
hilbert transform
#15121 opened Mar 21, 2023 by joglekara Loading…
@jakevdp
2
Defines the terms related to jax.Array.
#15110 opened Mar 20, 2023 by copybara-service bot Loading…
[xla-next][jax][sparse][mlir] implement jax.concatenate_p
#15108 opened Mar 20, 2023 by copybara-service bot Loading…
1
expose compiler_options on compile()
#15103 opened Mar 20, 2023 by copybara-service bot Loading…
Add padding option "SAME_LOWER" for ticket https://github.com/google/jax/pull/14990
#15102 opened Mar 20, 2023 by copybara-service bot Loading…
Force non-native jax2tf serialization.
#15098 opened Mar 20, 2023 by copybara-service bot Loading…
@gnecula
Support extrapolation in jnp.interp
pull ready Ready for copybara import and testing
#15086 opened Mar 20, 2023 by shoyer Loading…
1
@jakevdp
10
[1/n] store embedded tf.graph to stablehlo.custom_call
#15073 opened Mar 18, 2023 by copybara-service bot Loading…
@gnecula
explicitly convert jnp.var scalar normalizer to float (from int)
pull ready Ready for copybara import and testing
#15069 opened Mar 17, 2023 by mattjj Loading…
1
10
Internal change.
#15054 opened Mar 17, 2023 by copybara-service bot Loading…
Add xla_extension APIs for getting stablehlo serialization versions
#15040 opened Mar 16, 2023 by copybara-service bot Loading…
@gnecula
ProTip! Mix and match filters to narrow down what you’re looking for.