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
fix-notebooks2
d18f9a72
·
open and resave notebooks in colab
·
Jul 10, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jet
3e29c9fb
·
add initial jet tests
·
Aug 03, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
issue1236
merged
d700716e
·
add option to disable rank-promotion broadcasting
·
Aug 24, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
side-effects
a965b8d1
·
experimenting with effects
·
Aug 28, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
staxperiments
8f2c83a3
·
experiments on stax
·
Aug 29, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
solvers
698a259c
·
sketch of root w/ parameterized solvers
·
Aug 30, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
issue1297
merged
c760b05f
·
update jaxlib version in readme
·
Sep 02, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
update-notebooks
merged
a0095e34
·
reviewer comments: no 'install' text needed
·
Sep 02, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/staxperiments
af9a0d58
·
[WIP] final-style unzip for key-split-based NNs
·
Sep 13, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
ode-tests
217bdcf4
·
factor out ode tests into tests/ode_test.py
·
Sep 18, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
fixed-point
a0b92e09
·
experiments with a fixed_point implementation
·
Sep 18, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/safe-mul-pow-rhs
6d542e5a
·
use _safe_mul for _pow_jvp_rhs
·
Oct 03, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/scan-consts
ab282bd7
·
optimize scan with constant-across-time outputs
·
Oct 05, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/scan-consts-wip
18d47d60
·
wip, yikes
·
Oct 05, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/debug-strong-zeros
2f928e71
·
wip
·
Oct 13, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
travis-conda-build-issue
merged
0eea9885
·
try following jupyterlab/jupyter-renderers#212 to fix travis
·
Oct 16, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
issue1486
merged
585cefc8
·
document pmap with pytrees, fixes #1486
·
Oct 16, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jb/staxperiments-wip
ee2a92e0
·
pretty happy with this!
·
Oct 22, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
device-put-replicated
7c8f1ea3
·
Revert flawed jitted constant replicated device_put draft.
·
Oct 31, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
tweaks
76a1bb0c
·
tweaks to save precious vertical space
·
Nov 02, 2019
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
Prev
1
2
3
4
5
6
…
15
Next