research scientist @ Google Brain
-
Google
- San Francisco
- people.csail.mit.edu/~mattjj
Highlights
- Pro
- 24 discussions answered
Block or Report
Block or report mattjj
Report abuse
Contact GitHub support about this user’s behavior. Learn more about reporting abuse.
Report abuse661 contributions in the last year
Contribution activity
June 2022
Created 2 commits in 1 repository
Opened 1 pull request in 1 repository
google/jax
1
open
Reviewed 8 pull requests in 1 repository
google/jax
8 pull requests
- [dynamic-shapes] initial support for dynamic shape typechecks
-
Add
discharge_stateto convert stateful jaxprs into pure jaxprs - Add stateful for loop primitives
- Add an undocumented method on jit() functions to clear the function c…
- [x64] Make TPU svd compatible with strict type promotion
- [x64] make lax_numpy_test compatible with strict dtype promotion
- jnp.unique: improve error when run under JIT
- Enable debug print in xmap




