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
-
Google
- San Francisco
- Sign in to view email
- people.csail.mit.edu/~mattjj
1,364 contributions in the last year
Contribution activity
May 2020
Created a pull request in google/jax that received 7 comments
- Add support for 8- and 16-bit output in _random_bits
- Make lax.sort support tuple arguments using a variadic sort.
- Implement .view() method of jax.numpy arrays
- Avoid recompilation of rolled loops in threefry2x32.
- Fix sign error in examples.
- Modify linspace so that endpoints equal the inputs.
- Support axis_index_groups in allreduce collectives
- Implement np.digitize
- An implementation of id_tap and id_print using outfeed
- Improve JAX test PRNG APIs to fix correlations between test cases.
- Instantiate zeros in jet.
- Fixed a few more places where device commitment was lost.
- Add support for in_axes=None (but not out_axes, or in_axes>0) to pmap
- Reapply #2017 (Allow shapecheck of PixelCNN++), fixing #2245
- Fix test flakiness in autodiff tests for min/max type functions