The Wayback Machine - http://web.archive.org/web/20201206133028/https://github.com/google/jax/pulls
Skip to content

/ jax

New pull request New
136 Open 3,058 Closed
136 Open 3,058 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
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
Add support for linear algebra ops on ROCm using rocBLAS/rocSolver
cla: yes
#5115 opened Dec 5, 2020 by inailuig • Draft
Add support for AMD GPU's (ROCm)
cla: yes
#5114 opened Dec 5, 2020 by inailuig • Draft
1
Implement linear_ramp pad mode in jax.numpy.pad
cla: yes pull ready
#5110 opened Dec 5, 2020 by minoring • Approved
3
Add checking dimension of window_strides in shape rule
cla: yes
#5109 opened Dec 5, 2020 by minoring
[jax2tf] Fix bug in conversion of bfloat16 to tensor
cla: yes pull ready
#5108 opened Dec 5, 2020 by jakevdp • Approved
1
@gnecula
3
Cleanup: remove unnecessary rng_factory boilerplate from linalg_test.py
cla: yes pull ready
#5103 opened Dec 4, 2020 by jakevdp • Approved
Support mapping a single logical axis to multiple mesh axes
cla: yes
#5099 opened Dec 4, 2020 by apaszke
Update TF/TB versions for profiling in documentation.
cla: yes
#5090 opened Dec 3, 2020 by bchetioui
2
lax.convert_element_type: always return DeviceArray
cla: yes pull ready
#5085 opened Dec 2, 2020 by jakevdp • Approved
2
Ensure lax.convert_element_type preserves concrete inputs
cla: yes
#5075 opened Dec 2, 2020 by jakevdp
2
[jax2tf] Add testing for the conversion of (almost) all the remaining untested primitives
cla: yes pull ready
#5066 opened Dec 1, 2020 by bchetioui • Approved
3
all_to_all: Add optional axis_index_groups argument
cla: yes pull ready
#5059 opened Dec 1, 2020 by jpuigcerver • Approved
13
Expose sharded_jit, PartitionSpec, and with_sharding_constraint in jax.experimental namespace.
cla: yes pull ready
#5038 opened Nov 28, 2020 by skye
2
X32 tests: fail on dtype warnings
better_errors cla: yes pull ready
#5018 opened Nov 25, 2020 by jakevdp • Approved
3
Warn on host round-trip caused by jnp.array()
cla: yes
#5015 opened Nov 25, 2020 by jakevdp • Draft
Add vector-valued gradient example to quickstart
cla: yes
#4948 opened Nov 19, 2020 by gmittal
Improve pmap caching
cla: yes
#4923 opened Nov 17, 2020 by apaszke
Extract cloud TPU Colab set-up to a re-usable helper.
cla: yes
#4911 opened Nov 16, 2020 by aslanides
1
ci-build.yaml: mypy jax failing on the current version of Python
cla: yes
#4897 opened Nov 15, 2020 by cclauss
1
Add logging for pmap input (re)sharding
cla: yes
#4895 opened Nov 13, 2020 by jekbradbury
Cloned from CL 333372978 by 'g4 patch'.
cla: no
#4892 opened Nov 13, 2020 by copybara-service bot
1
DOC: Add description how to view pytree definition
cla: yes
#4856 opened Nov 10, 2020 by Baschdl
Fix batch polymorphism for take and gather
cla: yes pull ready
#4855 opened Nov 10, 2020 by gnecula
5
WIP: propagate weak types through unary and binary ops
cla: yes
#4850 opened Nov 10, 2020 by jakevdp • Draft
add gpu determinism note
cla: yes
#4824 opened Nov 7, 2020 by mattjj • Approved
4
ProTip! Adding no:label will show everything without a label.
You can’t perform that action at this time.