[D] Ideal deep learning library

From researcher perspective: - What do you miss in libraries like PyTorch or TensorFlow? - What could be improved?

Some possible examples: - The way how autodiff works - Debugging features - Working with axes, einops - Something that just feel awkward, inconvenient or incomplete

I would very much appreciate it if you could share your thoughts on this.

deep learning learning library machinelearning