-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operati…
-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operation…
-
Many tensor operations support the interface to take already allocated outputs, but the operations allocate memory for the output if the output passed as an argument is not allocated. The operations s…
-
What is the result of argmin_reduce and argmax_reduce when multiple dimensions are reduced?
For example, consider the tensor:
> 3.2 4.5 1.3
> 2.7 1.4 3.2
> 9.1 2.3 1.4
If we argmax_reduce d…
-
```otbuser@6dc23c62c23f:/home/data/decloud$ python3 models/train_from_tfrecords.py \
--training_record ./input/TFRecords/Cameroun/train/ \
--valid_records ./input/TFRecords/Cameroun/valid/ \
…
-
## 🚀 Feature
Make streams used for NCCL operations configurable
## Motivation
I've noticed that PyTorch distributed module has introduced P2P send and receive functionality via NCCL (which is…
wjuni updated
1 month ago
-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operation…
-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operation…
-
### 🐛 Describe the bug
The problem arises under the following circumstances (see below for repro):
- Model is compiled (fullgraph=True)
- Within the forward pass, a NamedTuple is instantiated
- …
-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operation…