ROCm/jax

Pull Requests Commits

shoyer committed 5 years ago

65ea8a8c

shoyer committed 5 years ago

666b4a80

Mark xfail tests

shoyer committed 5 years ago

71e413d3

Lint and mypy fixes

shoyer committed 5 years ago

35c975f0

Add another test case

shoyer committed 5 years ago

638dc489

Add tree rule for cconcatenate

shoyer committed 5 years ago

f5633a1b

add a simpler vmap(tree_call) test

shoyer committed 5 years ago

66f784d3

Simpler integration tests for vmap and jvp

shoyer committed 5 years ago

a77fac73

Use linear_util for all treevec transforms

shoyer committed 5 years ago

a0df0b1b

revert all changes to other files

shoyer committed 5 years ago

428ea0f8

Merge branch 'master' into undo-tree

shoyer committed 5 years ago

944a28a5

Move flattree into jax.experimental.treevec

shoyer committed 5 years ago

ab5c33ea

Merge pull request #3425 from google/document-ravel-pytree

mattjj committed 5 years ago

Verified 269da0ae

mattjj committed 5 years ago

159a61b2

add docstring to ravel_pytree

mattjj committed 5 years ago

ae9df752

factor autodiff and vmap tests out from lax_test

froystig committed 5 years ago

4a836ff8

reuse commonly-typed input/output values when joining partially evaluated conditional branches

froystig committed 5 years ago

ddea95e6

Bump XLA version. (#3424)

hawkinsp committed 5 years ago

Verified dd040de1

Merge pull request #3418 from fehiepsi/poissontype

mattjj committed 5 years ago

Verified 05fde0f1

Skip known invars and outvars in JaxprTrace.process_call (#3242)

apaszke committed 5 years ago

Verified 2dac55a0

allow scalar input in poisson sampler

fehiepsi committed 5 years ago

b680c994

Merge pull request #3413 from google/issue3412

mattjj committed 5 years ago

Verified d55c0ede

Change build badge from Travis to Github actions. (#3414)

hawkinsp committed 5 years ago

Verified ae60654d

supress warnings in test

mattjj committed 5 years ago

c6dcbe21

mattjj committed 5 years ago

a8e9e4f7

make jnp.array(x, copy=True) copies device buffers

mattjj committed 5 years ago

1a433620

Small edit to documentation. (#3406)

hawkinsp committed 5 years ago

Verified 04c9b327

Switch CI builds from Travis to Github actions (#3409)

hawkinsp committed 5 years ago

Verified 58f8329a

Improve error message when passing an invalid dtype. (#3405)

tomhennigan committed 5 years ago

Verified c7a021e8

Add num_partitions argument to sharded_jit (#3392)

skye committed 5 years ago

Verified 2f41a251

Older