Code

[WIP] call backend JAX bindings#74

Code

Open

mariogeiger wants to merge 6 commits intomainNVIDIA/cuEquivariance:mainfrom jax-bindingsNVIDIA/cuEquivariance:jax-bindingsCopy head branch name to clipboard

+293-19

Commits

Commits on Jan 23, 2025

• Implement frontend to call backend JAX bindings

  mariogeiger

  committed
  1b0af24

  Copy full SHA for 1b0af24

Commits on Jan 24, 2025

• Merge branch 'main' into jax-bindings

  mariogeiger

  committed
  69cf172

  Copy full SHA for 69cf172

• add logger info messages

  mariogeiger

  committed
  fe279c5

  Copy full SHA for fe279c5

• rename

  mariogeiger

  committed
  b5c7be8

  Copy full SHA for b5c7be8

Commits on Jan 28, 2025

• Add optional 'name' parameter to tensor product functions and update usage

  mariogeiger

  committed
  761e1c5

  Copy full SHA for 761e1c5

Commits on Jan 29, 2025

• profile_and_select_implementation

  mariogeiger

  committed
  06ffd6e

  Copy full SHA for 06ffd6e