ROCm/jax

Pull Requests Commits

Fix some bugs in _reshape_sharded_device_array (#2732)

skye committed 5 years ago

Verified 0e29bd4b

Temporarily make ShardedDeviceArray.__init__ optionally accept old si… (#2730)

skye committed 5 years ago

Verified fe8d1d1b

Update XLA. (#2733)

hawkinsp committed 5 years ago

Verified 14618d1a

Implement numpy.linalg.multi_dot (#2726)

jakevdp committed 5 years ago

Verified f9a3aed0

Add type hint to fix pytype error. (#2727)

skye committed 5 years ago

Verified 7a61aea5

Allow ShardedDeviceArrays to represent arbitrary data shardings. (#2142)

skye committed 5 years ago

Verified 07571ae4

Add a dynamic type check that the value returned by an XLA translation rule is an XlaOp. (#2723)

hawkinsp committed 5 years ago

Verified 87d95908

Explicitly build specific CUDA capabilities. (#2722)

hawkinsp committed 5 years ago

Verified 821193b2

Fixup complex values and tol in tests for jax.scipy.linalg.sparse.cg (#2717)

shoyer committed 5 years ago

Verified 8fa707af

Remove unused textwrap

danieldjohnson committed 5 years ago

f4f67b06

Add support for `mul`

danieldjohnson committed 5 years ago

9e429907

Merge branch 'master' into index_sugar

danieldjohnson committed 5 years ago

ab0a9031

Modify syntax to `x.at[idx].set(y)` and similar.

danieldjohnson committed 5 years ago

72593f04

Plumb precision argument into convolution in signal.py (#2715)

hawkinsp committed 5 years ago

Verified 5baa59fe

Instantiate RNG in testcase instead of test harness in a few more places. (#2706)

skye committed 5 years ago

Verified be6c5cb3

Fix some test failures. (#2713)

hawkinsp committed 5 years ago

Verified 1298e9e8

Merge pull request #2652 from jacobjinkelly/morejets

shoyer committed 5 years ago

Verified db1f6941

Merge pull request #2686 from aslanides/types

mattjj committed 5 years ago

Verified 2953a2bb

Merge branch 'types' of https://github.com/aslanides/jax into types

aslanides committed 5 years ago

61a1e9ce

Use Tuple[int, ...] rather than Sequence[int] in jnp.ndarray shape annotation.

aslanides committed 5 years ago

810b6f82

Implement numpy fmin() & fmax() (#2711)

jakevdp committed 5 years ago

Verified a134cc45

Update np.linalg docs with missing funcitons (#2710)

jakevdp committed 5 years ago

Verified 70b97259

Add type annotations to optix. (#2687)

aslanides committed 5 years ago

Verified 1ac80d71

Merge branch 'master' into types

aslanides committed 5 years ago

Verified 69cc477d

Update semantics of to_dlpack. (#2707)

hawkinsp committed 5 years ago

Verified bbf7a432

Fix minor typo in cell (#2692)

langosco committed 5 years ago

Verified d6ab70c3

Make type of value_and_grad slightly more precise. (#2704)

hawkinsp committed 5 years ago

Verified 7f6ede1b

Fix packbits/unpackbits tests (#2702)

skye committed 5 years ago

Verified 4e9f6404

Update XLA. (#2703)

hawkinsp committed 5 years ago

Verified 5ba9b6b5

Merge pull request #2701 from jakevdp/doc

mattjj committed 5 years ago

Verified 9c00550c

Newer Older