4 ms·
jax itself is just pip installable, so this should be drop-in. CUDA is it's own headache, and that's where mamba (now pixi[1]) really shines. [1]: https://pixi
by sooheon 2y ago
jax itself is just pip installable, so this should be drop-in. CUDA is it's own headache, and that's where mamba (now pixi[1]) really shines.
[1]: https://pixi.sh/ https://pixi.sh/
- insane_dreamer 2y agolooking at the uv docs it seems it doesn't support conda packages. That's a non-starter for us as we need both conda and pypi (some packages are in one, some in the other). So we'd probably look at pixi as a possible replacement for conda.
- maleldil 2y agoIf you really need conda, then you're on a _different_ nightmare than standard Python packaging.
- insane_dreamer 2y agoindeed we are; but CUDA+Jax, which we need, isn't properly supported on pypi
- insane_dreamer 2y agoso UV and PIXI are both nice in that they provide per-project environments (though using conda envs works just fine for us), but they don't solve the actual python packaging problem in that they still depend on either the pypi or conda packages (and neither support both)
- maleldil 2y agoYes. They are tools to manage the user side of packaging (i.e. installing, managing versions, locking, environments) but can't do anything to fix the problematic ecosystem. That is a much harder problem to solve.