ROCm/jax

Pull Requests Commits

add back note about block structure

mattjj committed 5 years ago

4d7e03e4

mattjj committed 5 years ago

47fb29fd

iterate on jax.hessian docs

mattjj committed 5 years ago

3a27213b

Implement nanargmin-max and add tests (#2398)

yurodiviy committed 5 years ago

Verified 56f6294e

add spacing to numpy.gradient (#2545)

Eduardo Pignatelli committed 5 years ago

Verified 2a0637a1

Document how jax.hessian and pytrees interact. (#2705)

hawkinsp committed 5 years ago

Verified ae6a3fe9

fix sort_key_val return type annotation, docstring

levskaya committed 5 years ago

Verified e599a254

Updated README wrt. new features for Stax. (#2862)

dynamicwebpaige committed 5 years ago

Verified 2611fd2d

Qr complex jvp fix (#2872)

Jamie Townsend committed 5 years ago

Verified f6e90603

Fix definition of qr primitive to return only the upper triangular part of r. (#2870)

hawkinsp committed 5 years ago

Verified e287f98c

Clarify that `grad` requires arguments to be differentiated to be of inexact type. (#2712)

hawkinsp committed 5 years ago

Verified 0dbbc27b

Merge pull request #2853 from levskaya/topkjvp

levskaya committed 5 years ago

Verified ca4e396e

Add top_k jvp and batching rules

levskaya committed 5 years ago

dddad2a3

Change isinstance test in xla_bridge.py to not explicitly name xla_client.Backend. (#2868)

hawkinsp committed 5 years ago

Verified c0023f44

Correct the order of .format arguments in vjp wrapper (#2866)

apaszke committed 5 years ago

Verified 5fe6b069

Add population_count primitive to lax (#2753)

Jamie Townsend committed 5 years ago

Verified 75617be8

Remove unused `ispure` method (#2781)

chr1sj0nes committed 5 years ago

Verified 2d96cfb2

refactor ode tests, add scipy benchmark (#2824)

jacobjinkelly committed 5 years ago

Verified cc0e9a31

Fix chi-squared tests in random_test.py (#2847)

shoyer committed 5 years ago

Verified e6df98de

Add pmap_shard_device_array_benchmark. (#2864)

skye committed 5 years ago

Verified 4b033433

check step size is greater than zero (#2857)

jacobjinkelly committed 5 years ago

Verified b277b55d

Update jaxpr.rst (#2859)

Jamie Townsend committed 5 years ago

Verified 283393f7

Simplify _odeint_rev (#2832)

samuela committed 5 years ago

Verified 5da74d4b

Add precision only arguments (#2850)

abhishekshrm53 committed 5 years ago

Verified 3736278f

split testDetGradOfSingularMatrix into corank=1,2 (#2845)

mattjj committed 5 years ago

Verified 0a5cc090

Fix typo in docstring for _cofactor_solve (#2844)

dpfau committed 5 years ago

Verified ffaf417b

Custom derivative for np.linalg.det (#2809)

dpfau committed 5 years ago

Verified 02b3fc5a

Enable some tests that now pass. (#2841)

hawkinsp committed 5 years ago

Verified 4e020cc4

Remove some tests for Jaxlib versions older than the minimum. (#2840)

hawkinsp committed 5 years ago

Verified f9806b35

Replace uses of xla_client.Buffer.from_pyval() with backend.buffer_from_pyval(). (#2839)

hawkinsp committed 5 years ago

Verified a6093d81

Older