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
cleaner-tie-in
b590df16
·
give tie_in a custom partial-eval rule
·
Nov 09, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/partial-eval-scan
9b296f8d
·
wip
·
Nov 11, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
explicit-embedding
fe5e4043
·
prototype an explicit embedding API (for a forthcoming pure_jit)
·
Nov 14, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/tagging
f4f6f163
·
fix tag of compile
·
Nov 14, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/readme-jaxlib
merged
5fa68774
·
Bump jaxlib version in README
·
Nov 17, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/broadcast-tie-in
7bee52e9
·
add tie_in to broadcast_arrays
·
Nov 17, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/sinc-double-where
merged
1817cab0
·
Use double-where trick to avoid NaNs in grad(sinc)
·
Nov 20, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
multi-host-pmap-fix
d68b9087
·
fix multi-host pmap, disambiguate nrep
·
Nov 20, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rvjp
036b5b58
·
Clean up prints an asserts
·
Nov 21, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jit-of-multi-host-pmap
ef95cf49
·
raise an error on jit-of-multi-host-pmap
·
Nov 26, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/scoping
07043f97
·
tagging tracer + collect/inject transforms, take 2
·
Dec 03, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
dynamic-experiments
66b182b7
·
wip
·
Dec 07, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
cloud_tpu_colabs
514a3a3e
·
Add Cloud TPU Colabs landing page.
·
Dec 14, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
lazy-sublanguage-eager-fusion
2d2958c9
·
fuse lazy expressions into eager computations
·
Jan 08, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
speed-up-hips
7c76e66d
·
removing stray selects doesn't help much
·
Jan 17, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
custom-transforms3-root
10b311e6
·
add back in comment
·
Jan 20, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
remove-x64-mode
020c4090
·
Fix dtype in scoring matching example
·
Jan 27, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
array-literals
64664379
·
improve the amount of test context
·
Feb 12, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
issue2248
08d07199
·
make device_get use onp.asarray, not .copy()
·
Feb 19, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
issue2263
0f115add
·
add population_count primitive (needs new jaxlib)
·
Feb 19, 2020
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
Prev
1
2
3
4
5
6
7
…
15
Next