jakevdp
Follow
Pinned
1,613 contributions in the last year
Less
More
Contribution activity
October 2022
Created 5 commits in 1 repository
Opened 9 pull requests in 1 repository
google/jax
4
open
5
merged
- [typing] overloads for jnp.linalg.svd & jnp.linalg.qr
- [typing] add type annotations to lax.linalg functions
- Call _check_arraylike for jnp.linalg & jnp.fft functions
- [typing] add types for jax.numpy.polynomial
- [typing] add annotations to jax.numpy.linalg
- [typing] add annotations to numpy.fft
- jax.jacobian: propagate function signature to transformed function
- Add types to jax/_src/numpy/util.py
- [typing] add full annotations for lax_numpy setops
Reviewed 7 pull requests in 1 repository
google/jax
7 pull requests
- Migrate api_test, lax_numpy_test, and lax_vmap_test to
- prod & sum: promote integers to default type rather than always 64
- [typing] add annotations to numpy.fft
- Implement jax.scipy.stats.truncnorm
- Bump styfle/cancel-workflow-action from 0.10.0 to 0.10.1
- Fix book links
- Add an internal jtu.sample_product test decorator.
Answered 4 discussions in 1 repository
google/jax
google/jax
-
raise ConcretizationTypeError when use fori_loop
This contribution was made on Oct 5
-
Why does custom pytree node produce node-in-node for `jax.jacfwd()`?
This contribution was made on Oct 4
-
How to enable type promotion warning?
This contribution was made on Oct 4
-
How are variable size pytrees jitted when variable sized arrays need to be a static argument?
This contribution was made on Oct 3