All pull requests

Pull requests

• made main branch work with both cuda and cpu-only

  #51·felipeasimos opened May 3, 2026·

  ·

• Convert codebase to JAX using Pallas for CUDA.

  #30·sguada opened Dec 16, 2024·

  ·

• Update difflogic_kernel.cu

  #25·Venalicius233 opened Nov 1, 2024·

  ·
  2

• bugfix: max number of unique connections

  #24·ligerlac opened Oct 15, 2024·

  ·

• Make CUDA extensions optional

  #22·ligerlac opened Oct 15, 2024·

  ·

• bugfix: recognize single layer as first and last

  #21·ligerlac opened Oct 14, 2024·

  ·

• make training kernels faster by using shared memory

  #15·mahdizaferanchi opened Jan 20, 2024Contributor·

  ·
  2