mattjj Follow

mattjj Follow

Matthew Johnson mattjj

Follow

research scientist @ Google Brain

1.3k followers · 129 following · 959

Google
San Francisco
people.csail.mit.edu/~mattjj

Achievements

Achievements

Highlights

Pro
12 discussions answered

Organizations

Block or Report

Block or report mattjj

Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse

Popular repositories

autodidact Public

A pedagogical implementation of Autograd

Jupyter Notebook 655 75
pyhsmm Public

Python 502 164
svae Public

code for Structured Variational Autoencoders

Python 320 92
pybasicbayes Public

Python 137 57
pyslds Public

Python 75 27
my-oh-my-zsh Public

Shell 74 15

783 contributions in the last year

Learn how we count contributions

Less More

Contribution activity

October 2021

Created 18 commits in 1 repository

google/jax 18 commits

Created 1 repository

mattjj/jax Python Oct 6

Created a pull request in google/jax that received 9 comments

rewrite jax.checkpoint (aka jax.remat), leave old implementation for now

The new jax.checkpoint is currently engaged whenever we pass an explicit policy. To get a "remat everything" decorator, one should write @partial(j…

+377 −23 • 9 comments

Opened 17 other pull requests in 1 repository

google/jax 17 merged

remat: fix regression of broke calling convention Oct 16
fix jax.checkpoint dce logic Oct 15
keep dropvar binders in call_partial_eval_custom_rule Oct 15
document axis_name in the vmap docstring Oct 14
make saved_residuals utility work w/ literals Oct 14
don't automatically use new checkpoint implementation Oct 14
checkpoint_name for checkpoint policies by name Oct 14
add more grad-of-jit/pmap caching tests Oct 13
remove ShapedArray.__len__ Oct 8
rbg_split and rbg_fold_in: use vmap for fewer HLOs Oct 8
update test of dlpack error message Oct 8
improvements to RBG PRNG Oct 7
fix rng_bit_generator translation rule Oct 5
lower rng_bit_generator using a BitcastConvertType Oct 5
pjit custom prngkey test Oct 3
don't use tree_map for pjit arg checking Oct 3
add experimental RngBitGenerator ("RBG") PRNG Oct 2

Reviewed 15 pull requests in 1 repository

google/jax 15 pull requests

Created an issue in google/jax that received 2 comments

names of jitted functions not shown in repr

In [1]: import jax.numpy as jnp In [2]: jnp.dot Out[2]: <CompiledFunction at 0x7f9fc7213c80>

2 comments