NVIDIA/cuEquivariance

SkillsMP は NVIDIA/cuEquivariance から 3 件の skill を収集しています。skill を開くとソースと詳細を確認できます。

NVIDIA のプロフィール GitHub で見る

記録された最新のソース活動: 2026年8月10日 21:07

SkillsMP カタログ更新: 2026年8月30日 18:55

収集済み skills: 3

GitHub スター: 423
GitHub フォーク: 39

このリポジトリの skills

データサイエンティストソフトウェア開発者

2 件の職業カテゴリ · 100% 分類済み

skill

職業分類

説明

更新

cuequivariance-jax

データサイエンティスト

Execute equivariant polynomials in JAX using segmented_polynomial (naive/uniform_1d), the ir_dict workflow with IrDictPolynomial and dict[Irrep, Array], and Flax NNX layers (IrrepsLinear, SphericalHarmonics, IrrepsIndexedLinear). Use when writing JAX code…

原文の言語: 英語

2026-08-10

cuequivariance-torch

ソフトウェア開発者

Execute equivariant tensor products in PyTorch using SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear), high-level operations (ChannelWiseTensorProduct, FullyConnectedTensorProduct, Linear, SymmetricContraction, SphericalHarmonics, Rotation), and…

原文の言語: 英語

2026-07-21

cuequivariance

データサイエンティスト

Define custom groups (Irrep subclasses), build segmented tensor products with CG coefficients, create equivariant polynomials and IrDictPolynomials, and use built-in descriptors (linear, tensor products, spherical harmonics). Use when working with…

原文の言語: 英語

2026-05-04

収集済み skill 3 件中 3 件を表示しています。