NVIDIA / NVIDIA/cuEquivariance

Conda packages for cuequivariance-ops-*

Open
#218 1 comment 1 reaction 0 assignees View on GitHub

Nobody has claimed this yet.

feature request
Dominant language
Python
Stars
433
Forks
42
PR merge metrics
No merged PRs in 30d

Description

I was wondering if you would consider creating a conda package for the cuequivariance-ops-(torch|jax)-cu(12|13) libraries. Optimally, for me would be if they could also be published on conda-forge, but that might be difficult given the license. Maybe there is another public channel that could be used.

I am currently part of a team that is trying to create a conda packaged version of the OpenFold-3 model. Having all the cuequivariance dependencies in conda(-forge) would help us immensely in this endeavour. The base cuequivariance libraries are already available on conda-forge. It is only the -ops libraries that are missing.

I was also trying to and would be happy to create the packages myself, but at least with grayskull this turned out to be difficult as there is (intentionally) no source distribution published on PyPI.

Of course, as I do not have access to the source, I do not know how difficult the build process for such a package would be. At the same time, given that the libraries have been packaged for PyPI, I would hope that the translation is rather straight forward (e.g., with the help of grayskull). Naturally, I completely understand if this is not a priority at this time.

Thank you for the library and your consideration 🙂

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by inspecting the PyPI packages for cuequivariance-ops-(torch|jax)-cu(12|13), the existing conda-forge packages for the base libraries, and grayskull's output. Check whether the package artifacts and license permit a conda recipe or another public channel. Done means the requested Torch and JAX CUDA 12 and 13 variants are buildable and published through an agreed channel.

Written by the indexing model from the issue text.

Assessment

Tech stack
anaconda, python, pytorch
Domain
build-system, release
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.