Code

feat: to/from PyTorch JaggedTensor#3246

Code

Closed

maxymnaumchyk wants to merge 21 commits intoscikit-hep:mainscikit-hep/awkward:mainfrom maxymnaumchyk:maxymnaumchyk/add-to-from-jaggedtensor-functionsmaxymnaumchyk/awkward:maxymnaumchyk/add-to-from-jaggedtensor-functionsCopy head branch name to clipboard

+372

Commits

Commits on Sep 17, 2024

• add new to_jaggedtensor function

  maxymnaumchyk

  committed
  9e28cb1

  Copy full SHA for 9e28cb1

• update __init__.py file

  maxymnaumchyk

  committed
  14b09d2

  Copy full SHA for 14b09d2

• add imports

  maxymnaumchyk

  committed
  70a2630

  Copy full SHA for 70a2630

• style: pre-commit fixes

  pre-commit-ci[bot]

  committed
  9feb758

  Copy full SHA for 9feb758

• add style changes

  maxymnaumchyk

  committed
  4674108

  Copy full SHA for 4674108

• Merge remote-tracking branch 'origin/maxymnaumchyk/add-to-from-jaggedtensor-functions' into maxymnaumchyk/add-to-from-jaggedtensor-functions

  Show description for c0da418
  maxymnaumchyk

  committed
  c0da418

  Copy full SHA for c0da418

• add style changes

  maxymnaumchyk

  committed
  000e10a

  Copy full SHA for 000e10a

Commits on Sep 18, 2024

• check if fbgemm is installed

  maxymnaumchyk

  committed
  b820b51

  Copy full SHA for b820b51

• correct checking if fbgemm is installed

  maxymnaumchyk

  committed
  ade482e

  Copy full SHA for ade482e

• add cuda support

  maxymnaumchyk

  committed
  fc97f37

  Copy full SHA for fc97f37

• add comments

  maxymnaumchyk

  committed
  0f48a9a

  Copy full SHA for 0f48a9a

• style changes

  maxymnaumchyk

  committed
  5e68ab5

  Copy full SHA for 5e68ab5

• fix tests

  maxymnaumchyk

  committed
  2cdc0db

  Copy full SHA for 2cdc0db

Commits on Sep 19, 2024

• leave out the "padded" and "keep_regular" arguments

  maxymnaumchyk

  committed
  5d2eaad

  Copy full SHA for 5d2eaad

• add error for non existing backend

  maxymnaumchyk

  committed
  81fd78a

  Copy full SHA for 81fd78a

• *better* handle backend stated by user

  maxymnaumchyk

  committed
  4244642

  Copy full SHA for 4244642

Commits on Sep 20, 2024

• separately handle cupy arrays

  maxymnaumchyk

  committed
  86d5922

  Copy full SHA for 86d5922

• delete "backend" argument since the same can be done with ak.to_backend()

  maxymnaumchyk

  committed
  b316181

  Copy full SHA for b316181

• add new "to_jaggedtensor" function

  maxymnaumchyk

  committed
  d1d723b

  Copy full SHA for d1d723b

• style fixes

  maxymnaumchyk

  committed
  940b466

  Copy full SHA for 940b466

Commits on Sep 25, 2024

• Merge branch 'main' into maxymnaumchyk/add-to-from-jaggedtensor-functions

  ianna

  authored
  c68321c

  Copy full SHA for c68321c