google/xls 379
XLS: Accelerated HW Synthesis
llvm opt fuzzer and bounded exhaustive test generator
fork of llvm-stress
Mirror of official clang git repository located at http://llvm.org/git/clang. Updated every five minutes.
Mirror of official compiler-rt git repository located at http://llvm.org/git/compiler-rt. Updated every five minutes.
This repo contains Microsoft compiler-tests to validate Windows platform particulars.
TensorFlow documentation
C++ standards drafts
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
PR opened google/jax
We support half16 on TPU
pr created time in a month
push eventmajnemer/jax
commit sha 6f5afa3494238b2c0c1a8c5289bede0e1bc3a4f7
Remove type restrictions We support half16 on TPU
push time in a month
PR opened google/jax
We support s8, u8, s16, u16, half16 on TPU
pr created time in a month
push eventmajnemer/jax
commit sha 6bab85b8d3e347e65d6b5a8bee00951a9b3600ef
Remove type restrictions We support s8, u8, s16, u16, half16 on TPU
push time in a month
PR opened google/jax
dynamic_slice and dynamic_update_slice work with S16 types on TPU.
pr created time in a month
push eventmajnemer/jax
commit sha c8801fbd9bbb2d6563f7186d26e857c2ed3866ff
Remove type restrictions dynamic_slice and dynamic_update_slice work with S16 types on TPU.
push time in a month
issue commentgoogle/jax
random uniform with dtype bfloat16 crashes on TPU backend
I think this should work now.
comment created time in a month
PR opened google/jax
pr created time in 2 months
push eventmajnemer/jax
commit sha f8bab4ae7c6293148be219b3ce0fd963a9c955cf
update version and changelog for pypi
commit sha 66ba734882daf8206e41c96495a9ea3c53082bb1
Add note to docs describing how pytree arguments work. (#3284) Addresses #3095. I'm not sure if we wanna link to this from API docstrings. This also subsumes the original pytrees notebook.
commit sha b998044ffed2f36f683c9dce4fbd2914d5b7de44
Add np.polyadd (#3261)
commit sha 0db57cb541d58f223cabb02bed4aca51a3d876c3
Fix validation code in lax.conv (#3279)
commit sha d1dbf7c7d8216158024697abc9db7d3576b22ebd
Implement mask for some primitives + jit. (#2922) * Implement mask for slice, conv, pad, transpose, where * Remove tentative mask(jit) * Add explanatory comment to dot_general masking rule * Rm reshape from select masking rule * Rm unnecessary check from lax slice abstract_eval rule * Revert to standard indentation in masking_test.py * Begin simplifying masking tests * Finish drafting masking check function * More progress simplifying tests * Add conv masking in batch dim * Finish fixing up tests * Revert to old API, making out_shape compulsory again * More efficient conv masking rule * Tidy up masking_test imports * Check that out tree is preserved by masking * fix flake errors Co-authored-by: Jamie Townsend <[email protected]> Co-authored-by: Jamie Townsend <[email protected]> Co-authored-by: Matthew Johnson <[email protected]>
commit sha c77c0838fea4a4673754c5348a56d00bcb7f8bad
deflake jax.numpy and add to flake8 check (#3312)
commit sha 5ad9feda5f74a88e053a33c5f5330044186b128f
Fix handling of infeed token inside sharded_jit (#3313)
commit sha 4f5547dd85596bff775cc1ed42c013321833a8b6
Don't AD through max-subtraction in softmax (#2260) * Don't AD through max-subtraction in softmax * Also stop-grad the max in logsumexp
commit sha dc4c9f045007959b6bcd9c1c97b3f958f09fc706
change cond primitive to an indexed conditional with multiple branch functions in the core: * bind and check cond primitive in indexed form * rewrite abstract evaluation rule * rewrite translation rule * rewrite partial evaluation rule * rewrite batching rule * rewrite JVP rule * rewrite transpose rule * update jaxpr typechecker * update pretty printer * update outfeed-usage check * update reference jaxpr in cond jaxpr test * update reference regexes in HLO test in experimental modules: * update host_callback rewriter * update loops expression builder * generalize tf_impl rule
commit sha 6015a2a6893af98195a0b121056527e1124ab76f
introduce lax.switch
commit sha bd3cab9768370a58d172a5bdb2d39de776189a49
update jaxpr doc to reflect lax.switch and indexed cond
commit sha c49bb754543f89fc44bcec2ab4b7824f3b869be0
update changelog with lax.switch
commit sha afa9276f0869305afe12cbaec88fe3fb535de807
Implement jax_to_tf.scan (#3307)
commit sha 71f1c5cafeab50f0358aa194f0a03a0666e4b4db
Refactoring of jax_to_tf tests: (#3262) (#3308) * Moved control-flow tests into their own file * Added a helper module tf_test_util, with a helper function ConvertAndCompare * Used self.assertAllClose instead of numpy.testing.assert_all_close because the former iterates over lists and tuples (and is standard in other JAX tests) * Used @parameterized.named_parameters for parameterized tests, for nicer test names.
commit sha c04dea1c84b657d014412a04a5312e7e525b7501
Begin implementing mask(jit)
commit sha dfe3462746d701b08f3d1ee814534f228d2fa199
Add device_put_p abstract_eval rule
commit sha 0f0032727b68e3fc07164c3505ad5c80c9c08503
Implement MaskTrace.post_process_call
commit sha 38d483737d0e35f77ed79fc279574b4b2dc46937
Fix x64 test
commit sha 9c0a58a8e774171e5e465bd81d2fad481c5264fc
add float dtype checks to random.py (#3320) fixes #3317
commit sha 45444363449ee651386d283235661c4f18339a47
Improve error when zero-sized arrays passed to convolve
push time in 2 months
issue commentgoogle/jax
[jax2tf] Incorrect out of bound index handling for lax.dynamic_slice
Can scatter ignore out of bound indices?
comment created time in 3 months
push eventmajnemer/jax
commit sha 6653cdbb3b5d5329847c97469fd158519be2417b
Fix lax_reference's round for edge case inputs - round(8388609) would compute trunc(8388609 + 0.5) == 8388610. Fix this by not modifying sufficiently inputs. - round(0.499999970198) would compute trunc(0.499999970198 + 0.5) == 1.0 Fix this by explicitly special casing the first float before 0.5.
push time in 3 months