jakevdp
Follow
jakevdp
Follow
Achievements
Achievements
Highlights
- 102 discussions answered
Pinned Loading
1,476 contributions in the last year
Less
More
Contribution activity
November 2021
Created 53 commits in 4 repositories
Created a pull request in google/jax that received 13 comments
Document jax.lax.Precision
Fixes #8447 View generated documentation here: https://jax--8450.org.readthedocs.build/en/8450/jax.lax.html#jax.lax.Precision Since it's not possib…
+80
−40
•
13
comments
Opened 39 other pull requests in 4 repositories
google/jax
1
open
31
merged
1
closed
- [x64] make dtypes.dtype accept a single value.
- jnp.quantile: explicitly raise error for complex input
- CI: disable mypy false positive
- [sparse] fix batched version of BCOO.sum_duplicates
- Remove stray print statements
- [sparse] support type promotion in CSR/COO matmul
- [sparse] bcoo_[sp]dot_general: require inputs to have the same type
- jnp.unique: ensure that output dtype is not affected by fill_value
- [sparse] spdot_general: implement more cases
- [sparse] add BCOO.sum_duplicates() with nse option
- [sparse] ensure shapes are represented as tuples
- [sparse] re-enable bcoo_spdot_general test
- [sparse] add dtype assertions to several cusparse wrappers
- [x64] Add test of jnp.arange() dtype corner case
- [x64] clean up usage of dtypes.dtype
- [x64] jnp.array: improve type inference testing
- [x64] set scalar defaults to 64-bit types
- jnp.percentile: use full precision for 64-bit inputs
- [sparse] improve bcoo validation
- Properly handle bfloat16 in jnp.load()
- refactor jax.numpy.meshgrid & improve validation
- [sparse] bcoo_dot_general: support contraction along batch dimensions
- jax_check_tracer_leaks: add warning about debuggers
- colab_tpu.setup_tpu: allow user specification ot tpu driver version
- [sparse] register BCOO as a vmappable
- Some pull requests not shown.
altair-viz/altair
3
merged
altair-viz/altair_viewer
2
merged
altair-viz/altair_saver
1
open
Reviewed 15 pull requests in 2 repositories
google/jax
14 pull requests
- [shape_poly] Update arange and image_resize to support shape polymorphism
- Increment minimum jaxlib version in mypy precommit hook.
- [sparse] add BCOO.sum_duplicates() with nse option
- [x64] Add test of jnp.arange() dtype corner case
- Update bcoo_dot_general GPU translation rule.
- fix for translation to cudaDataType
- add Lowered and Compiled to the jax public package
- update version and changelog for pypi
- Generate a better error message if [pv]map receives no array arguments
- Document jax.lax.Precision
- [sparse] register BCOO as a vmappable
- Update coo_matvec and coo_matmat docstrings.
- [sparse] add autodiff & transpose rules for csr primitives
- Implement hermitian=... option on jax.numpy.linalg.svd.
altair-viz/altair
1 pull request
Created an issue in altair-viz/altair that received 17 comments
Version 4.2 release candidate?
Let's get a version 4.2 release candidate together! In the past, I always just directly created new releases, but it seems Altair has become pretty…
17
comments
Opened 4 other issues in 1 repository
Answered 14 discussions in 1 repository
google/jax
google/jax
- Grad of constant?
- Poor Jax Sparse Performance with GPU
- Benchmarks within JIT
- lax.scan with generator
- Issues with saving and pickling bfloat16 array
- vmap with reduction
- scan with dynamic_slice seems slow
- Vmap over distinct (but static) slice sizes
- Using a HashableArrayWrapper inside JIT'ted function fails
- Way around BCOO batch dimension contraction in example?
- jax.random.permutation doesn't fully replace jax.random.shuffle
- Sparse tensordot with sparse outputs?
- SVD dimensions
- Jax `grad` complains about ints but my data contains no ints