Arthur McCray
Results
2
comments of
Arthur McCray
@vfdev-5 I see what you mean. From what I can tell, it looks like it would be possible to enable vmapping once only, as at the end of the day...
Has there been any progress on this? I'm also in need of DDP for complex-valued NNs, and I am wondering if there are plans for this issue to be resolved.