ROCm/jax

Pull Requests Commits

did somebody say grad of pjit?

mattjj committed 7 years ago

f2f23c5d

pjit tests working again

mattjj committed 7 years ago

75849f2b

mattjj committed 7 years ago

821731ce

partial progress

mattjj committed 7 years ago

08dc6994

partial progress

mattjj committed 7 years ago

549a1f2e

jaxtuple flattening problems...

mattjj committed 7 years ago

893ec515

add temp axis name class

mattjj committed 7 years ago

2c447b98

tweak batching.moveaxis behavior

mattjj committed 7 years ago

9fa80779

basic nested pjit works

mattjj committed 7 years ago

6efb7b1c

add basic pjit_nesting.py work file

mattjj committed 7 years ago

41c98ddc

working on nested pjit, misc bug fixes

mattjj committed 7 years ago

da620a8e

Merge pull request #305 from hawkinsp/rename

hawkinsp committed 7 years ago

Verified 5e37d2a4

Reexpose reduce_window_shape_tuple since it has external users.

hawkinsp committed 7 years ago

5517347c

Prefix most rules in lax module with underscores to improve generated doc readability.

hawkinsp committed 7 years ago

09201c72

Merge pull request #303 from hawkinsp/minmax

hawkinsp committed 7 years ago

Verified 66c7a424

Fix complex64/complex128 confusion in test harness.

hawkinsp committed 7 years ago

89727e43

Fix gradient for `np.amin` and `np.amax`.

hawkinsp committed 7 years ago

fb659e22

Merge pull request #300 from alexalemi/rev_batching

mattjj committed 7 years ago

Verified 670f14a2

Add newly added Numpy functions to docs.

hawkinsp committed 7 years ago

8d52ee89

Merge pull request #297 from alexalemi/flip{up,lr}

hawkinsp committed 7 years ago

Verified 785ada21

more test cases

Alex Alemi committed 7 years ago

3558fa99

Add batching rule for rev.

Alex Alemi committed 7 years ago

a9b221a1

Add support for negative indices and tests of rank 3 tensors.

Alex Alemi committed 7 years ago

4c37d249

Add `np.cumproduct` alias for `np.cumprod`.

hawkinsp committed 7 years ago

edce0f0a

Merge pull request #298 from hawkinsp/cumsum

hawkinsp committed 7 years ago

Verified 4e84bf8a

Work around missing ReduceWindow cases in current Jaxlib release.

hawkinsp committed 7 years ago

a3f0da50

Implement np.{cumsum,cumprod,nancumsum,nancumprod}.

hawkinsp committed 7 years ago

26f85310

Alex Alemi committed 7 years ago

2edc509c

Add and numpy functions.

Alex Alemi committed 7 years ago

89e3b3a1

Merge pull request #294 from google/stop-gradient

mattjj committed 7 years ago

Verified 7eb8579c

Newer Older