google / jax

Watch 222
Star 9.4k
Fork 767

Code
Issues 474
Pull requests 109
Discussions
Actions
Security
Insights

Code
Issues
Pull requests
Discussions
Actions
Security
Insights

Labels 19 Milestones 1

New pull request New

109 Open 2,386 Closed

Author

Filter by author

Label

Filter by label

Use alt + click/return to exclude labels.

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

Use pytree from xla_client.

cla: yes

#4063 opened Aug 14, 2020 by jblespiau • Changes requested

[jax2tf] Added tests for reduce_window translation

cla: yes

#4062 opened Aug 14, 2020 by SIben • Draft

[jax2tf] Fix bfloat16 bug in select_and_gather_add conversion.

cla: yes

#4058 opened Aug 14, 2020 by SIben

Add `in_parts` and `out_parts` optional arguments `jax.xla_computation`.

cla: yes

#4055 opened Aug 14, 2020 by rsepassi

Add a fast C++ jit codepath.

cla: yes

#4051 opened Aug 13, 2020 by jblespiau

Note flush-denormal-to-zero behavior in docstring of nextafter.

cla: yes

#4049 opened Aug 13, 2020 by mathisgerdes • Approved

Support mask(pmap)

cla: yes

#4046 opened Aug 13, 2020 by j-towns

Regroup Jit tests into the same class.

cla: yes

#4045 opened Aug 13, 2020 by jblespiau

Add some tests to jax_jit.

cla: yes

#4041 opened Aug 13, 2020 by jblespiau

Grad of int

cla: yes

#4039 opened Aug 12, 2020 by LenaMartens • Draft 2 of 6

omnistaging on by default

cla: yes

#4038 opened Aug 12, 2020 by mattjj • Draft

Add early check for pmap inputs larger than number of local devices.

cla: yes

#4037 opened Aug 12, 2020 by skye • Approved

Fixing weird behavior in segment_sum when num_segments is None.

cla: yes

#4034 opened Aug 12, 2020 by alvarosg

[jax2tf] Clean up code for XlaGather, experimental_compile not necessary

cla: yes

#4030 opened Aug 12, 2020 by gnecula

Added initial implementation of numpy equivalent for trim_zeros to jax

cla: yes

#4027 opened Aug 12, 2020 by gsp-27

Add GMRES to `jax.scipy.sparse.linalg`

cla: yes

#4025 opened Aug 11, 2020 by gehring • Draft 0 of 4

custom_jvp/vjp closure issues, just experimenting for now

cla: yes

#4008 opened Aug 10, 2020 by mattjj • Draft

Initial version of gmap

cla: yes

#4006 opened Aug 10, 2020 by apaszke

Initial version of vmap collectives

cla: yes

#4005 opened Aug 10, 2020 by apaszke

WIP: allow jax objects to be represented by multiple buffers

cla: yes

#3983 opened Aug 6, 2020 by jakevdp • Draft 3 of 6

Add NumPy backend

cla: yes

#3923 opened Jul 31, 2020 by JuliusKunze • Approved

Clarify error message in scan transpose

cla: yes

#3917 opened Jul 30, 2020 by jekbradbury • Approved

Add test for dtype coverage of jax.numpy ufuncs

cla: yes

#3913 opened Jul 30, 2020 by jakevdp • Draft

WIP: exploring weakly-dtyped device arrays

cla: yes

#3903 opened Jul 29, 2020 by jakevdp • Draft

Make PyTreeDef pickleable for Multiprocessing

cla: yes

#3868 opened Jul 27, 2020 by bionicles

Previous 1 2 3 4 5 Next

Previous Next

ProTip! Follow long discussions with comments:>50.