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
skip-dlpack-tests-on-jaxlib-version
4a19de60
·
added test skips based on jaxlib version
·
Oct 21, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
4008-fixes
default
protected
79e3db55
·
fixes on #4008 (thanks @apaszke)
·
Oct 21, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
test_338274969
48e26c28
·
Internal change
·
Oct 22, 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
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
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
docfix-jit
9fc53b6e
·
fix jit docstring
·
Oct 27, 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
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
handle-dunder-array-classes
4aa10cab
·
deflake
·
Oct 28, 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
test_340511784
042d37d5
·
Internal change
·
Nov 04, 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_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
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
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
pdot
483d0717
·
minor refactoring useful for pdot
·
Nov 25, 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
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
issue1583
a9cb42d4
·
try making 0D DeviceArrays hashable
·
Dec 02, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
Prev
1
…
4
5
6
7
8
9
10
11
12
…
15
Next