jakevdp
Follow
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 upPinned
1,599 contributions in the last year
Contribution activity
November 2020
Created 39 commits in 4 repositories
Created a pull request in google/jax that received 5 comments
PRNGSeed: ensure JIT invariance for valid inputs.
This is a less drastic version of #4821 that aims only for JIT invariance for jax.random.PRNGKey, leaving dtype invariance breakages untouched. The…
+32
−16
•
5
comments
Opened 38 other pull requests in 2 repositories
google/jax
31
merged
4
open
2
closed
- X32: don't warn when builtin type passed as dtype
- X32 tests: fail on dtype warnings
- Warn on host round-trip caused by jnp.array()
- Call asarray() rather than array() to avoid host round-trips.
- Support equal_nan parameter in jnp.allclose
- jnp.array(): avoid unnecessary numpy array copy
- Fix random doc
- Support 'where' argument in jax.numpy reductions
- Make jnp.gradient() call signature match np.gradient
- Cleanup: remove obsolete TODOs
- Support 'initial' argument in jax.numpy reductions
- Increase test coverage for reductions.
- jax.numpy: match numpy arguments for reductions
- Add initial changelog for jax v0.2.7
- update version & CHANGELOG for jax v0.2.6
- Remove noisy warnings for var() of int
- Make dtype an explicit argument to nan-reductions
- fix mypy error
- Fix mypy
- Support shape parameter in jax.numpy.*_like()
- lax_numpy: fix handling of out keyword in wrapped methods
- Update in1d() TODO
- Fix dtypes.is_weakly_typed
- TST: add explicit tests of PRNGKey seed handling.
- Deprecation: Raise a TypeError when indexing with a non-tuple sequence
- Some pull requests not shown.
altair-viz/vega_datasets
1
merged
Reviewed 6 pull requests in 1 repository
google/jax 6 pull requests
- Add complex types to gradient of slogdet
- fix grad-of-switch weak type mismatch test
- Change the DeviceArray error from ValueError to RuntimeError when del…
- use `typematch` rather than `==` in cond transpose assertion
- jax.scipy.signal functions for n-dim arrays
- Add documentation to several functions in jax.lax.linalg.
Created an issue in google/jax that received 2 comments
jit fails for large uint64 inputs
Repro: import jax.numpy as jnp from jax import jit, random, config config.update('jax_enable_x64', True) x = jnp.iinfo(jnp.uint64).max print(jnp.ui…
2
comments
