apaszke
Follow
Create your own GitHub profile
Sign up for your own profile on GitHub, the best place to host code, manage projects, and build software alongside 50 million developers.
Sign up
Pinned
479 contributions in the last year
Contribution activity
September 2020
Created a pull request in google/jax that received 1 comment
Add a prototype implementation of binomial checkpointing
Example: import jax import jax.numpy as jnp from jax.experimental.binomial import checkpoint_binomial @checkpoint_binomial def f(x): for i in range(8…
+47
−0
•
1
comment
- Improve defunctionalization of case statements
- Stop capturing loop indices when reifying the tangent function
- Differentiation through inactive cases + misc improvements in AD
- Try using GitHub Actions for CI instead of Travis
- Defunctionalize through case expressions
- Another bunch of AD improvements
- Don't linearize inactive expressions
- Flesh out transposition
- Flesh out linearization
- Expand the C API and Python bindings
- A batch of AD fixes and improvements
- A prototype of Python bindings
- Make it possible to use ptxas instead of the CUDA driver for SASS compilation
- Improve DCE and inliner
- Inline table builders (`for`s) with a single use site
