Create your own GitHub profile
Sign up for your own profile on GitHub, the best place to host code, manage projects, and build software alongside 50 million developers.
Sign up
Popular repositories
-
Forked from google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python 2
-
Forked from flapjack/flapjack
Monitoring notification routing + event processing system. For issues with the Flapjack packages, please see https://github.com/flapjack/omnibus-flapjack/
Ruby
-
-
-
Forked from gatolteam/gatol
Rails Backend for Game-A-Thon of Learning, a platform for learning games
Ruby 1
498 contributions in the last year
Contribution activity
June 2020
Created a pull request in google/jax that received 2 comments
Refined the forcing of annotation processig in doc building
Re-implemented the insertion of annotations during doc building
* the line is inserted before the first import. Otherwise, Sphinx will ignore the m…
- [jax2tf] Add support for custom JVP/VJP
- [jax2tf] Add special case for translation of lax.gather to tf.gather.
- Disable the workaround to prevent expansion of type aliases
- [jax2tf] Control on which TF device the TF converted code runs.
- [jax2tf] Fixed bug with TF silently using 64-bit operations in 32-bit mode.
- [jax2tf] Added special case for tf.pad.
- [jax2tf] fix the too-early use of tf.constant
- [jax2tf] separated primitives_test.py
- [jax2tf] Fixed the handling of `core.unit` in control-flow primitives.
- [jax2tf] Fixed bug in argument type promotion for dynamic_update_slice
- [jax2tf] Added test about primitive coverage
- Renamed experimental/jax_to_tf to experimental/jax2tf
- Fix forgotten references to ad.zero in notebook
- Moved shift_right implementation from tfxla to jax_to_tf
- Fix imports for jax_to_tf tests
- Refactor the jax_to_tf tests to separate the primitive test harness
- Revert "Make ad_util.zero a class that carries avals (similar to Unde…
- A couple of ad_util.zero were missed in #3222
- Commented-out the literal jaxpr checks in host_callback
- Extract JAX primitive test harness from lax_test.py.
- Refactor jax_tf_tests
- Implement jax_to_tf.scan
- Reverting jax_to_tf (scan and tests)
Created an issue in google/jax that received 3 comments
Leaking trace stack global state in presence of exceptions
I ran into a situation when the core.trace_state was left in a bad state after a failing test. Here is a simple repro:
def test_leaking_trace_stac…