Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
mirrors
ROCmSoftwarePlatform
jax
Repository
Branches
Overview
Active
Stale
All
add_norocm_all_tests_patch
c9b578f8
·
more no-rocm tags
·
Aug 06, 2021
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
check_test_teardown
a2266fc2
·
add 1 second delay between tests
·
Aug 04, 2021
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
convert-element-type-old-dtype
8ed7be49
·
make convert_element_type_p not require old_dtype
·
Dec 04, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
issue1583
a9cb42d4
·
try making 0D DeviceArrays hashable
·
Dec 02, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
zero-fixes2
8fef9e8a
·
clean up symbolic zero handling in transpose rules
·
Nov 29, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
debug-nans-stores
7266fcbd
·
when resetting stores in nan debugging, check None
·
Nov 26, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
pdot
483d0717
·
minor refactoring useful for pdot
·
Nov 25, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
switch-grad-weak-type-test
76d43035
·
fix grad-of-switch test under x64 mode
·
Nov 17, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
log-resharding
d79f48b5
·
Add logging for pmap input (re)sharding
·
Nov 14, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
test_337964891
14b4a6cc
·
Cloned from CL 333372978 by 'g4 patch'.
·
Nov 13, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
gpu-determinism-note
535d984f
·
add gpu determinism note
·
Nov 08, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
test_340511784
042d37d5
·
Internal change
·
Nov 04, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
typo
57d3fd87
·
add missing space in error message
·
Oct 30, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
handle-dunder-array-classes
4aa10cab
·
deflake
·
Oct 28, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
static-argnums-immutable
1da36943
·
mention static_argnum values should be immutable
·
Oct 28, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
simpler-select-jvp
7a73e99e
·
simplify select jvp
·
Oct 27, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
docfix-jit
9fc53b6e
·
fix jit docstring
·
Oct 27, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
traceback-register
fa1d7ab5
·
move `wraps` from `api_util` to `util` to avoid cyclic dependencies
·
Oct 27, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
custom-gradient-wrapper
3c6cdcfc
·
add jax.custom_gradient wrapper for jax.custom_vjp
·
Oct 24, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
make-custom-vjp-bwd-nones-more-robust
3e697fb7
·
make the custom_vjp bwd None handling more robust
·
Oct 23, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
Prev
1
…
3
4
5
6
7
8
9
10
11
…
15
Next