I’ve recently came along pixi, it is for nix like what vscode for other more powerfull obscure code editors
https://ericmjl.github.io/blog/2024/8/16/its-time-to-try-out-pixi
Is something this nice (hassle free UX-wise) with same resulst possible within the nix ecosystem Potentitally devenv. But creating a working env with both pytorch and jax with gpu support and some extra python packages feels much harder (if at all possible) with it.
I do belive nix has many benifits to the conda/pixi world (more packages, more flexibility). But the UX, ease of use and approachability is definetely lacking 
Are there any clean examples of python-based dev-shells with modern DL frameworks and tooling that are easy to use and not hacky?
Some notable ones I know (but still not there)