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
Popular repositories
-
-
Forked from tensorflow/tensorflow
Computation using data flow graphs for scalable machine learning
C++
-
-
-
1,160 contributions in the last year
Contribution activity
September 2020
- Enable fast TPU LU decomposition for complex types.
- Enable complex-valued Cholesky decomposition tests on TPU>
- Lower LU decomposition to a custom TPU implementation for float32 types.
- Revert "Revert "Delete batching.last. (#4148)""
- Change lax_linalg.lu to return a permutation representation of the pa…
- Update scale_and_translate to take an explicit spatial_dims.
- Disable CPP Jax jit tests.
- Improve segment_sum stability by k-way summation.
- Expose jax.image.scale_and_translate
- Bump TF hash to get an upstream LLVM GCC fix.
- Reorder nocuda/cuda build to fail early.
- Update install_cuda script to specify cublas.