jakevdp
Follow
Pinned
1,619 contributions in the last year
We're celebrating 100 million developers!
Contribution activity
January 2023
Created 45 commits in 2 repositories
Created a pull request in google/jax that received 4 comments
DOC: add GPU & TPU installation tabs on index page
As discussed this morning. Live preview: https://jax--14047.org.readthedocs.build/en/14047/#installation
+18
−4
•
4
comments
Opened 44 other pull requests in 1 repository
google/jax
4
open
38
merged
2
closed
- DOC: fix autodidax
- DOC: clarify purpose of the jax-array doc
- DOC: organize content in advanced guide
- DOC: more guidance within getting started guide
- Link to JAXopt in example_libraries.optimizers
- CI: fix mypy jaxlib version
- Use __slots__ in Jaxpr & ClosedJaxpr
- CI: update flake8 to v6.0.0
- Raise an error when attempting to mutate Jaxpr objects
- [sparse] refactor tests to improve runtime
- [sparse] add a few missing methods to BCOO
- CI: update deprecated uses of set-output
- CI: adjust permissions for upstream-nightly build
- Reorganization among notes, user guides, advanced guides
- [sparse] update primitive coverage in module doc
- DOC: add JAX ecosystem links
- DOC: make API doc titles more uniform
- Fix excessive recompiles in lax.cond
- DOC: make grid placement responsive for smaller screens
- Sharp bits dynamic shapes
- Sharp bits: raise exceptions directly
- FAQ: fix discussion of marking self as static
- [typing] add missing attributes to jax.Array
- DOC: fix heading levels in GPU custom ops
- Fix finfo test for older numpy versions
- Some pull requests not shown.
Reviewed 23 pull requests in 1 repository
google/jax
23 pull requests
- DOC: clarify purpose of the jax-array doc
- Removed deprecated polar_unitary as per phawkins comment.
- DOC: organize content in advanced guide
- [sparse] add sparse.map_specified_elements primitive
- DOC: add JAX ecosystem links
- Make JAX Debugging and Profiling guides more visible
- Update lax_control_flow_test.py
- Fix headings in JAX Custom operations for GPUs and Building from source guides
- DOC: make grid placement responsive for smaller screens
- Add IO callback
- Add beginner, advanced, and contributors guide
- Added shape error checking for compute_fans
-
Migrate pytestmark usage to new
@jtu.pytest_mark_if_availabledecorator. -
Add
pjrt_c_api_unimplementedpytest marker to skip unsupported tests. - [docs] change donate_argnums FAQ link to rst format
- mgrid: Fix zero-length meshgrid
- DOC: add FAQ entry on converting a tracer to an array
- (add) add SmeLU activation function to nn.functions
- Update jax.xmap notebook title, make "xmap" tutorial more discoverable
- Fix broken link in readme
- add remat tutorial docs
- Fix typo in checkify guide.
-
[Typing] Add
Optionalto parameterboundaryofstft.
Created an issue in google/jax that received 4 comments
lax.cond leads to excessive compilation in JAX 0.4.1
Reported in #14032 (reply in thread). Minimal reproduction: import jax import jaxlib from jax import lax print(jaxlib.__version__) print(jax.__vers…
4
comments
Opened 6 other issues in 1 repository
google/jax
3
open
3
closed
Answered 17 discussions in 1 repository
google/jax
google/jax
-
How can I make my function bfloat16 compatible?
This contribution was made on Jan 27
-
What is the best way to get both the Jacobian and Hessian of a function?
This contribution was made on Jan 25
-
`value_and_hessian`?
This contribution was made on Jan 25
-
Interpreter idioms - staging to Jaxpr, vs. writing own tracers?
This contribution was made on Jan 23
-
`vmap` across arbitrary leading dimensions.
This contribution was made on Jan 17
-
Gradients of functions which accept `Pytrees` with arbitrary leaf types (how to ignore `Int`)?
This contribution was made on Jan 17
-
vmap on list?
This contribution was made on Jan 16
-
Passing array of random 'keys' in vmap
This contribution was made on Jan 12
-
JAX matmul (100x100) seems ~16x slower than numpy because no asynchronous behavior?!
This contribution was made on Jan 12
-
Using pmap with a sparse matrix
This contribution was made on Jan 10
-
Freezing parameters with `partial` function or `static_argnames` argument.
This contribution was made on Jan 10
-
JIT-compiled functions updating and returning non JAX types. Intended or not?
This contribution was made on Jan 7
-
jax.lax.scan using lists of arrays
This contribution was made on Jan 6
-
How to use lax.while_loop for array of logicals.
This contribution was made on Jan 5
-
lax.pdot() function status
This contribution was made on Jan 5
-
How to construct a 'jit'able function to delete an entry of an array
This contribution was made on Jan 4
-
Jax vmap, in_axes doesn't work if keyword argument is passed
This contribution was made on Jan 1