mattjj
Follow
mattjj
Follow
Achievements
Achievements
Highlights
- Pro
- 23 discussions answered
637 contributions in the last year
Contribution activity
May 2022
Created 17 commits in 2 repositories
Created a pull request in google/jax that received 6 comments
remove count attribute and total_ordering from core.Var
Originally we used the Var.count attribute to ensure Var instances were printed consistently regardless of context, even though only their object i…
+15
−78
•
6
comments
Opened 12 other pull requests in 1 repository
google/jax
11
merged
1
open
- add core.closed_call_p
- tweak mlir shape_tensor helper, fewer MHLO ops
- fix grad(..., argnums=-1), regressed in #10453
- improve partial_eval_jaxpr_custom
- make core_test.py pass with core.call
- fix ad_checkpoint.checkpoint vmap rule
- remove jnp.array case for handling buffers w/ aval=None
- [new-remat] add scan rule for new remat
- add xla_client._version checks for mhlo.ConstOp signature
- add more scan dce rule tests, fix issues
- [remove-units] remove units
- [remove-units] remove units from partial_eval.py
Reviewed 20 pull requests in 1 repository
google/jax
20 pull requests
- Enable colors when we are using a terminal
- add core.closed_call_p
- DOC: add references for haskell-style signatures
- Remove unnecessary __future__ imports
- improve partial_eval_jaxpr_custom
- Add "Sequencing side-effects in JAX" design note
- Fix _lax_max_taylor_rule for broadcastable inputs
- make core_test.py pass with core.call
- remove jnp.array case for handling buffers w/ aval=None
- Enable ordered effects in cond lowering
-
Enable ordered effects in the
condof a while loop - DOC: remove mentions of units
-
Enable AD rules for
debug_print -
Enable batching rule for
debug_print - Enable effect lowering for while/for/scan
- remove count attribute and total_ordering from core.Var
- Don't leak the keepalive in debug_callback lowering
- Initial debug print implementation
- add more scan dce rule tests, fix issues
- [remove-units] remove units from partial_eval.py