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 upHighlights
- Arctic Code Vault Contributor
Pinned
1,572 contributions in the last year
Contribution activity
July 2020
Created a pull request in google/jax that received 4 comments
WIP: x64: make Jax default dtypes 32-bit
This is step 1 in removing the x64 flag: jax should default to 32-bit types to facilitate working on accelerators.
+83
−72
•
4
comments
- Fix type mismatch in jet rule for abs
- [x64 deprecation] Create _np_array utility routine
- Change onp/np to np/jnp in docs & notebooks
- Cleanup: convert uses of 'import numpy as onp' in tests
- Cleanup: convert uses of `import numpy as onp` in library code
- lax_numpy: rename arguments to match numpy
- Fix compilation bug in histogram_bin_edges
- Remove unused private helper function
- fix flake error on master
- lexicographic sort_p: accept num_keys rather than comparator
- Remove some non-inclusive language
- Generalize lax.sort to support lexicographic sorts.
- Add _NOT_IMPLEMENTED attribute to jax.numpy (fixes #3689)
- Cleanup: record names in get_module_functions
- Cleanup: use test_util dtypes where possible
- Revert #3610 & #3684
- linalg_test: define test matrices lazily
- fix pmap test on GPU/TPU
- Add CI tests for numpy 1.16.4
- Add test of jax.numpy vs numpy call signatures
- tweak jnp.repeat not to use jnp on shapes
- Fix `jax.image._resize` function
- add clarification about jit inside indexing error message
- Error message and docstring updates RE: dynamic_slice
- Note in pmap docs that pmap compiles like jit.
- Don't move batch dimensions to start in jnp.einsum.
- Relax test tolerance for conv_general_dilated gradients.
- Use keyword arguments in einsum.
- Update scipy.ndimage.map_coordinates docstring
- Fix bug in jnp.repeat where for array repeats with zero entries (e.g. [1,0,3]) and axis=0 the zero entry was not skipped).
- Fix low probability (4/1000) flake of batching_test on GPU.
- A compilable version of np.repeat.
- Make jnp.take work for empty slices of empty arrays.
- Adding np.sort_complex within lax_numpy.py
- Add pshuffle to docs
- Improve np.intersect1d
- Implement complex convolutions on CPU and GPU.
- Implement np.intersect1d
- Fix a link to TensorBoard's profiler in Profiling JAX Programs
- fix bad pmap tests from #3675
- Make LU gradient work for low-rank matrices
- Add simple JAX API microbenchmarks.
- Remove a deprecated reference to testExamplesJaxprDoc in Understanding Jaxpr
- Adding support for array split
- Add np.trim_zeros numpy function

