ROCm/jax

Pull Requests Commits

Use double-where trick to avoid NaNs in grad(sinc)

jekbradbury committed 6 years ago

1817cab0

Bump jaxlib version to 0.1.33 and update WORKSPACE.

skye committed 6 years ago

b1888881

fix reference cycle in jaxpr tracing using weakrefs

mattjj committed 6 years ago

1fcebbaa

remove very old "what we're working on"

mattjj committed 6 years ago

Verified 5edda9e6

Merge pull request #1707 from google/jb/readme-jaxlib

jekbradbury committed 6 years ago

Verified 68c05331

bump version for pypi

mattjj committed 6 years ago

a3474fec

Implement the __pos__ operator on JAX arrays. (#1718)

hawkinsp committed 6 years ago

Verified 5c3b99d0

Check for None in indexer dtype check. (#1717)

hawkinsp committed 6 years ago

Verified f95e3e96

Add type check that indexers are integers or boolean values. (#1716)

hawkinsp committed 6 years ago

Verified 6cf2e4b8

Relax test tolerance for core_test jvp tests. (#1714)

hawkinsp committed 6 years ago

Verified d323431d

Avoid out-of-bounds dereference for arity-0 nodes. (#1713)

hawkinsp committed 6 years ago

Verified 9679a879

Change scalar promotion rules to prefer array types over scalar types. (#1709)

hawkinsp committed 6 years ago

Verified 42dd736a

Merge pull request #1706 from gnecula/loops

gnecula committed 6 years ago

Verified 397a244e

Implemented suggestions from code review.

gnecula committed 6 years ago

8ec6ea47

fix lax.scan notes in gotchas notebook

levskaya committed 6 years ago

Verified f8823595

Bump jaxlib version in README

jekbradbury committed 6 years ago

Verified 5fa68774

tweak test name (cf. #1704)

mattjj committed 6 years ago

063419ab

Chase Roberts committed 6 years ago

3978007b

Cast perm to tuple

Chase Roberts committed 6 years ago

979a8d30

Change test tolerance logic not to choose tolerance values based on f… (#1701)

hawkinsp committed 6 years ago

Verified bbf8129a

Improved documentation

gnecula committed 6 years ago

d549d44e

Fix tests for Python 2 and for X64

gnecula committed 6 years ago

64e186c3

An implementation of an experimental syntactic sugar for 'for' loops.

gnecula committed 6 years ago

d24c374d

Update XLA. (#1702)

hawkinsp committed 6 years ago

Verified 9b853a42

Drop protobuf dependency from `jax` package. It appears unused. (#1700)

hawkinsp committed 6 years ago

Verified 4fc76524

Move internal type-related functions into a new (internal) jax.types … (#1695)

hawkinsp committed 6 years ago

Verified f4aa5150

improve docs on shape/dtype loop stability

mattjj committed 6 years ago

3ac32713

Merge pull request #1697 from google/grad-argnums-error-message

mattjj committed 6 years ago

Verified dbf41348

improve grad error message without enough args

mattjj committed 6 years ago

728cb7fb

skip some tests on tpu

mattjj committed 6 years ago

be28700b

Older