ROCm/jax

Pull Requests Commits

add gpu determinism note

mattjj committed 5 years ago

535d984f

Merge pull request #4801 from bchetioui:jax2tf_named_call

a-googler committed 5 years ago

2f753630

Merge pull request #4788 from jakevdp:promotion-via-lattice

a-googler committed 5 years ago

556b1461

Merge pull request #4806 from jakevdp:lattice-doc

a-googler committed 5 years ago

4db3db9e

Do type promotion directly via lattice rather than pre-generating a table.

jakevdp committed 5 years ago

67c1c8c0

DOC: add lattice to type promotion doc

jakevdp committed 5 years ago

86f45353

[jax2tf] Systematically add the functions in xla.call_translations

bchetioui committed 5 years ago

c670ab33

Merge pull request #4807 from hawkinsp:docs

a-googler committed 5 years ago

8f1be202

Add documentation to several functions in jax.lax.linalg.

hawkinsp committed 5 years ago

2cf2c719

Merge pull request #4799 from hawkinsp:laxfix

a-googler committed 5 years ago

11d6f239

Fix max/min confusion in lax.py.

hawkinsp committed 5 years ago

79cf7d11

Merge pull request #4606 from bchetioui:quickdraw_tfjs

a-googler committed 5 years ago

bd98b060

Merge pull request #4794 from jakevdp:result-type-fiux

a-googler committed 5 years ago

5c9204a4

Merge pull request #4733 from qiuminxu:upstream_named_call

a-googler committed 5 years ago

4138639e

Merge pull request #4792 from jakevdp:cleanup-dtypes

a-googler committed 5 years ago

f49f626c

Cleanup: remove unused utility function

jakevdp committed 5 years ago

370f30c6

Move lax linear algebra routines into a jax.lax.linalg module.

hawkinsp committed 5 years ago

575a8e06

Merge pull request #4787 from jakevdp:fix-jax-type

a-googler committed 5 years ago

9d6d0274

Update lax_test.py

qiuminxu committed 5 years ago

430ba446

qiuminxu committed 5 years ago

3b315b86

Add named_call test

qiuminxu committed 5 years ago

17d684ee

Upstream named_call to jax

qiuminxu committed 5 years ago

70f03eb6

Fix: always canonicalize result_type

jakevdp committed 5 years ago

ee3a071b

Merge pull request #4791 from hawkinsp:laxcleanup

a-googler committed 5 years ago

3c14abc9

Merge pull request #4789 from hawkinsp:xla

a-googler committed 5 years ago

ca066b6a

Delete deprecated jax.lax.lax* modules which were previously private implementation details of jax.

hawkinsp committed 5 years ago

a3d5fe3c

[JAX] Move source_info_util into jax._src.

hawkinsp committed 5 years ago

7efc1dbc

hawkinsp committed 5 years ago

608cfde9

Fix issue in jax.dtypes._jax_type

jakevdp committed 5 years ago

16e01a17

Merge pull request #4786 from hawkinsp:import

a-googler committed 5 years ago

644de249

Older