Corey adams

[BUG] Fatal Python error: PyMUTEX_LOCK(gil->mutex) failed

21

Hi, I've got a package that uses pybind11 (it's awesome, by the way), and had a few users report the following crash. I've been able to reproduce it myself as...

Enable support for Intel XPU devices (AKA Intel GPUs)

1

## What does this PR do? This PR extends pytorch_lighting with support for Intel GPUs, as enabled with `intel_extension_for_pytorch`. With Intel's module, pytorch gains the `torch.xpu` module which is equivalent...

Unexpected behavior with JIT'ing allreduce

10

I've encountered a strange behavior when using mpi4jax that has, in retrospect, been plaguing me for months. I've managed to put it into a reproducer here: https://gist.github.com/coreyjadams/6640c00c3dc0f202989cc964b8995214 (UPDATE: Here is...

Enable opt-in autodetection of distributed configuration

8

This PR is in response to the discussion on #19409. It does the following: - First, it adds an additional cluster environment for jax.distributed that is based on autodetection of...

Proposal for generic mpi4py initialization of jax distributed module

11

`jax.distributed.initiallize()` works, without arguments, on several but not all common MPI / Slurm parallel job launchers. Unfortunately, the environment variables used in the `ompi_cluster.py` class are not standardized. I'd like...

XPU Installation Instructions Don't Work

43

### Describe the bug When I copy/paste the installation instructions for XPU, they fail: ``` ❯ python -m pip install torch==1.13.0a0+git6c9b55e intel_extension_for_pytorch==1.13.120+xpu -f https://developer.intel.com/ipex-whl-stable-xpu Looking in links: https://developer.intel.com/ipex-whl-stable-xpu ERROR: Could...