ROCm/jax

Pull Requests Commits

better test str message

fehiepsi committed 6 years ago

688d77f4

add some nonlinearity

fehiepsi committed 6 years ago

2836d03b

support ddof for var

fehiepsi committed 6 years ago

45c5bd4f

Add rmsprop_momentum optimizer (same as TF RMSProp)

ibab committed 6 years ago

628f87d3

add numpy.median

fehiepsi committed 6 years ago

98152d9d

Merge pull request #1073 from hawkinsp/deviceget

hawkinsp committed 6 years ago

Verified c41677fa

Merge remote-tracking branch 'upstream/master' into sort

fehiepsi committed 6 years ago

1b490fb5

add batching rule for lax.sort

fehiepsi committed 6 years ago

e1ee87b5

Add a pytreedef.flatten_up_to() method that flattens a PyTree only up to the structure of a PyTreeDef.

hawkinsp committed 6 years ago

38bffe9a

Merge pull request #1084 from hawkinsp/xla

hawkinsp committed 6 years ago

Verified 0533813b

hawkinsp committed 6 years ago

cf95ed87

Correct jax.numpy.pad signature

j-towns committed 6 years ago

Verified 47f9eedb

Merge pull request #1079 from google/fix-pad-transpose-symbolic-zeros

mattjj committed 6 years ago

Verified 2466b9b9

fix symbolic zero handling in _pad_transpose

mattjj committed 6 years ago

0600b738

Add corrcoef to jax.numpy docs.

hawkinsp committed 6 years ago

6434d9d0

Replace jaxlib import paths with jax.lib paths.

hawkinsp committed 6 years ago

00ca6a2f

Merge pull request #1074 from hawkinsp/pytree

hawkinsp committed 6 years ago

Verified a3501913

Address additional review comments.

hawkinsp committed 6 years ago

e8b42367

Add how to configure TF to GPU memory allocation doc.

skye committed 6 years ago

3461dfff

Add docs on GPU memory allocation.

skye committed 6 years ago

5b9849f1

move ode.py to jax/experimental

mattjj committed 6 years ago

b9a984f4

Merge pull request #1068 from duvenaud/ode-demo

mattjj committed 6 years ago

Verified 6c199fcd

Address review comments.

hawkinsp committed 6 years ago

3c3f01e6

Merge pull request #1071 from wyjw/master

hawkinsp committed 6 years ago

Verified 96dd2e66

Update lax_numpy.py

wyjw committed 6 years ago

Verified 4dcae5de

Update lax_numpy_test.py

wyjw committed 6 years ago

Verified 5b75a7c1

make test more robust

wyjw committed 6 years ago

Verified 22a6aaf9

Update lax_numpy_test.py

wyjw committed 6 years ago

Verified ea8597ed

Remove old xla_data_pb2 compatibility shim.

hawkinsp committed 6 years ago

d0644d6a

Add C++ implementation of pytree logic.

hawkinsp committed 6 years ago

510a9167

Newer Older