ROCm/jax

Pull Requests Commits

pretty happy with this!

jekbradbury committed 6 years ago

ee2a92e0

jekbradbury committed 6 years ago

0cbad476

jekbradbury committed 6 years ago

35f04c46

jekbradbury committed 6 years ago

5076a611

jekbradbury committed 6 years ago

74eb0d8c

jekbradbury committed 6 years ago

1de56881

back to a primitive

jekbradbury committed 6 years ago

e8c98155

more general/fewer primitives

jekbradbury committed 6 years ago

1c2e635f

[WIP] final-style unzip for key-split-based NNs

jekbradbury committed 6 years ago

c83245ae

experiments on stax

mattjj committed 6 years ago

85cf5587

Revert "Add a pylintrc to make it easier to use linter (#1442)"

skye committed 6 years ago

54807b42

Revert "Remove build_mac_jaxlib.yml for now to see if that's causing spurious workflows."

skye committed 6 years ago

2308d0f2

Merge pull request #1462 from google/jb/fix-primitive-metadata

jekbradbury committed 6 years ago

Verified d34d9eb1

add test for the case that failed

jekbradbury committed 6 years ago

245d29b4

fix XLA metadata for primitives with many args

jekbradbury committed 6 years ago

bc0e7976

Remove build_mac_jaxlib.yml for now to see if that's causing spurious workflows.

skye committed 6 years ago

59ae24e8

Bump jaxlib version to 0.1.31 and update WORKSPACE.

skye committed 6 years ago

d3fa506e

Merge pull request #1455 from gnecula/colabs

gnecula committed 6 years ago

Verified 6c6e90d0

Merge pull request #1457 from shoyer/readthedocs-yml

gnecula committed 6 years ago

Verified c6ac94f8

update to version 2 schema

shoyer committed 6 years ago

b7335be2

minor README formatting and grammar adjustment

froystig committed 6 years ago

cd29780f

Add a .readthedocs.yml file

shoyer committed 6 years ago

f30fc7c6

Merge pull request #1446 from joaogui1/selu

jekbradbury committed 6 years ago

Verified 2b46e0db

Increase readthedocs/nbsphinx timeout

gnecula committed 6 years ago

41457633

Merge pull request #1423 from gnecula/colabs

gnecula committed 6 years ago

Verified a5063ee6

joaogui1 committed 6 years ago

505caaae

Add a pylintrc to make it easier to use linter (#1442)

joaogui1 committed 6 years ago

a0bb2c0e

Merge pull request #1438 from google/jb/xla-debug-info

jekbradbury committed 6 years ago

Verified 88911199

remove source lines (caching makes them ~useless)

jekbradbury committed 6 years ago

4e6385f8

Fix typo in notebook JSON (#1441)

gnecula committed 6 years ago

c4e41995

Older