ROCm/jax

Pull Requests Commits

Update docs/notebooks/How_JAX_primitives_work.ipynb

gnecula committed 6 years ago

Verified 73144faf

Update docs/notebooks/How_JAX_primitives_work.ipynb

gnecula committed 6 years ago

Verified ef886c7f

Fix typo in initial_step_size d2 calculation (#1428)

Dan Weaver committed 6 years ago

f2052132

Merge pull request #1427 from trevorcai/doc

jekbradbury committed 6 years ago

Verified f04f06f5

Merge pull request #1436 from google/jb/ortho-init-type

jekbradbury committed 6 years ago

Verified b1120e4f

provide lax primitive + src line as XLA debuginfo

jekbradbury committed 6 years ago

59343b1a

Typo: Gradeints -> Gradients

fabianp committed 6 years ago

5201c363

Refactor pytree munging inside lax.root. (#1414)

shoyer committed 6 years ago

Verified e8bf56df

Update README to point to jaxlib-0.1.30

skye committed 6 years ago

7f98682f

fix missing delimiter in notebook source

froystig committed 6 years ago

db3b0dd0

Merge pull request #1425 from gnecula/colab_testing

shoyer committed 6 years ago

Verified d13599e8

more initializer aliases

jekbradbury committed 6 years ago

aa35f8b0

add initializer tests

jekbradbury committed 6 years ago

23f06b41

fix two typos in orthogonal init

jekbradbury committed 6 years ago

4210192d

Switched testing to using sphinx (part of the documentation generation)

gnecula committed 6 years ago

0fdf3773

Added support for testing Colab notebooks.

gnecula committed 6 years ago

57e39ce9

Bump jaxlib version to 0.1.30 and update WORKSPACE.

skye committed 6 years ago

2a8575bf

Merge pull request #1429 from google/jb/eig-debug-nans

jekbradbury committed 6 years ago

Verified 22e32516

Fix the title to show nicely in sphinx

gnecula committed 6 years ago

7bc0f70f

Addressed comments for the Colab.

gnecula committed 6 years ago

75c22360

unbreak debug_nans on jit

jekbradbury committed 6 years ago

51038457

test that debug_nans actually finds NaNs

jekbradbury committed 6 years ago

dd1e132e

fix config.update when absl is enabled

jekbradbury committed 6 years ago

d174fd66

address comments

jekbradbury committed 6 years ago

efc3a2c3

use setUp and tearDown

jekbradbury committed 6 years ago

be4e156a

add basic tests

jekbradbury committed 6 years ago

459bd969

fix debug_nans for multiple-result primitives

jekbradbury committed 6 years ago

ee3fe5f5

Move epsilon term in optimizers.rmsprop into sqrt

trevorcai committed 6 years ago

c18d765c

Merge pull request #1424 from necula01/documentation1

shoyer committed 6 years ago

Verified e3597384

Fixed links to Colabs in docs/notebooks

gnecula committed 6 years ago

6744389d

Newer Older