ROCm/jax

Pull Requests Commits

add docstring / reference doc link for axis_index

mattjj committed 6 years ago

fcc1e76c

Merge pull request #2527 from lucasb-eyer/patch-1

mattjj committed 6 years ago

Verified 614d39dc

Added lots of trivial jet rules.

duvenaud committed 6 years ago

ead80118

add new custom_jvp tests from #2500

mattjj committed 6 years ago

67283a08

Merge pull request #2533 from google/core-trace-type-annotations

mattjj committed 6 years ago

Verified bcc5191c

add type annotations to core.py tracing machinery

mattjj committed 6 years ago

f99720b7

Merge pull request #2532 from google/sharded-device-array-handlers

mattjj committed 6 years ago

Verified cdbcbb00

add ShardedDeviceArray to ad vspace op handlers

mattjj committed 6 years ago

1b597895

Make it more explicit that default JVP assumes |R

lucasb-eyer committed 6 years ago

Verified 415cde5b

Improve speed of LU decomposition on TPU. (#2526)

hawkinsp committed 6 years ago

Verified f371bfc0

Merge pull request #2501 from botev/isclose

mattjj committed 6 years ago

Verified 6d0810a4

Fix pmap_benchmark.py import (#2524)

skye committed 6 years ago

Verified 24bbd2bc

Merge pull request #2523 from sharadmv/tangent-fix

mattjj committed 6 years ago

Verified db0ac989

Instantiate zeros in _custom_vjp_call_jaxpr_jvp

sharadmv committed 6 years ago

0499b845

attempt to fix link formatting with nbsphinx

mattjj committed 6 years ago

Verified 42dbfd43

Merge pull request #2513 from google/temp-revert-2026

mattjj committed 6 years ago

Verified a4215f29

remove deprecation warnings

mattjj committed 6 years ago

3700fab7

temporarily revert parts of #2026 pending bug fix

mattjj committed 6 years ago

c3e3d480

fix derivatives reference (wrong Rudin!)

mattjj committed 6 years ago

Verified 32747476

improve ref to Tao's 3rd edition of Analysis I

mattjj committed 6 years ago

Verified fc0f875b

remove incorrect sentence in notebook

mattjj committed 6 years ago

Verified da9b5232

Added test for isclose.

botev committed 6 years ago

9ec0ac6e

Added FAQ entry about creating JAX arrays

gnecula committed 6 years ago

f88d49b4

Added a FAQ for impure functions

gnecula committed 6 years ago

86e3046e

Started a FAQ for JAX

gnecula committed 6 years ago

6f2f779a

Better error message for indexing with floats (#2496)

gnecula committed 6 years ago

Verified d1e8f43a

Merge pull request #2503 from google/issue2502

mattjj committed 6 years ago

Verified a4922ced

improve custom_jvp error messages, fixes #2502

mattjj committed 6 years ago

74c20509

Making isclose handle correctly infinite and NaN values.

botev committed 6 years ago

b05ac579

fix custom_transforms bug

mattjj committed 6 years ago

0cf84f92

Newer Older