Peter Hawkins
Peter Hawkins
(And you'll need to sync on top of head)
mypy is sad, see the lint_and_typecheck result.
I just reapproved the CI runs (unless you're a collaborator on the repository it needs approval to run CI — I'm surprised you're not, actually, let me know if you...
@benkirk You don't need to build JAX from source to use a custom NCCL. We'll use whichever `libnccl.so` we find in your `LD_LIBRARY_PATH`.
I wrote that bfloat16 extension (the TF/JAX one) targeting NumPy 1.16 when working on early TPUs. NumPy's type extension APIs have certainly advanced since then. So it's probably time someone...
The JAX repo bug is fine: in practice we are the ones working on this and that's the project we work in most. (Really this extension should be its own...
We don't support the PPC architecture ourselves and most likely don't have the engineer bandwidth to maintain such a build. But we wouldn't object if the community wanted to supported...
As to your specific question, I'd make sure that MKLDNN is disabled in the build (I believe there is an option to build.py for this.) I doubt MKLDNN works on...
@feifzhou I don't know how to solve your issue, but it looks to me like Bazel isn't understanding something about the location of the standard library headers on your system....
@f0uriest If you can share the output of the build, we might be able to suggest things to change. I'd speculate there are two or three things you'd need to...