Search results

New Jax Release (0.9.2) breaks quax's importing at runtime

Status: Open.

#111 In nstarman/quax;

· AXSaketh opened on Apr 10, 2026

BUG: tracer leaks

Status: Open.

#68 In nstarman/quax;

· nstarman opened on Jun 16, 2025

Parameter counting when quaxified layers are invoked

questionUser queriesUser queries

Status: Open.

#66 In nstarman/quax;

· neel04 opened on Jun 12, 2025

Using quax with jax.jvp

Status: Open.

#65 In nstarman/quax;

· nardi opened on Jun 9, 2025

Interest in benchmarking?

Status: Open.

#61 In nstarman/quax;

· nstarman opened on Apr 23, 2025

Problem in _Quaxify.__call__

questionUser queriesUser queries

Status: Open.

#58 In nstarman/quax;

· nstarman opened on Apr 14, 2025

Overloading jax operations to avoid undefined derivatives

questionUser queriesUser queries

Status: Open.

#35 In nstarman/quax;

· alleSini99 opened on Nov 12, 2024

Class methods vs. quaxified functions

questionUser queriesUser queries

Status: Open.

#33 In nstarman/quax;

· vadmbertr opened on Oct 15, 2024

scan primitive implementation missing

Status: Open.

#29 In nstarman/quax;

· colehaus opened on Sep 4, 2024

LoRA that doesn't require memory for zero gradients of the underlying matrices

questionUser queriesUser queries

Status: Open.

#28 In nstarman/quax;

· colehaus opened on Aug 28, 2024