ROCm/jax

Pull Requests Commits

fix tag of compile

jekbradbury committed 6 years ago

f4f6f163

fix batching rule for push_tag

jekbradbury committed 6 years ago

c6bfc7f8

jekbradbury committed 6 years ago

80467228

support random.split

jekbradbury committed 6 years ago

329a6edf

note in dict_util

jekbradbury committed 6 years ago

4b4e7e01

fix a couple typos

jekbradbury committed 6 years ago

7f2f1266

relax tie_in preference for first path

jekbradbury committed 6 years ago

d299edd6

add tagging tracer and collect/inject transforms

jekbradbury committed 6 years ago

95f051fe

Update WORKSPACE to include TF build fix.

skye committed 6 years ago

c4f6e54a

Merge pull request #1667 from google/cleanups

mattjj committed 6 years ago

Verified 9392824f

[doc] Note that building jaxlib from source isn't always necessary (#1654)

trevorcai committed 6 years ago

340d82e9

Revert "Update bazel min version to 0.26.0."

skye committed 6 years ago

44ccca05

Update bazel min version to 0.26.0.

skye committed 6 years ago

807a1958

Merge pull request #1674 from levskaya/linspace2

mattjj committed 6 years ago

Verified b566cc2a

allow tokens in op-by-op by calling into _xla_callable_args

mattjj committed 6 years ago

6cd995e3

Bump jaxlib version to 0.1.33 and update WORKSPACE.

skye committed 6 years ago

84437839

Add TPU Driver to jaxlib (#1673)

henrytansetiawan committed 6 years ago

ddbdcfb9

fix degenerate case behavior of linspace

levskaya committed 6 years ago

350630fd

temporarily revert #1658 due to TFP test failures

mattjj committed 6 years ago

0d053f0e

improve DeviceArray.__repr__ with onp.array2string

mattjj committed 6 years ago

938eb20b

fix DeviceArray repr (for google internal test)

mattjj committed 6 years ago

c4101c56

Update jaxlib version in README.md. (#1669)

hawkinsp committed 6 years ago

Verified 3d4bb775

Merge pull request #1658 from levskaya/linspace

levskaya committed 6 years ago

Verified 07201502

linspace, logspace, geomspace jittable and differentiable in start and stop args

levskaya committed 6 years ago

03287304

add missing attribute in optix (for google tests)

mattjj committed 6 years ago

d77cf175

fix xla.lower_fun and jax.xla_computation

mattjj committed 6 years ago

6f47ac00

simplify [p]xla.py by in-lining single-caller function

mattjj committed 6 years ago

4aaf325d

Merge pull request #1666 from google/xla-cleanups

mattjj committed 6 years ago

Verified 70612530

rename TestSpec -> CallSpec to avoid warning

mattjj committed 6 years ago

46db509b

simplify xla op metadata, DeviceArray repr

mattjj committed 6 years ago

5ec07491

Older