Search results

BUG: tracer leaks

Status: Open.

#68 In patrick-kidger/quax;

· nstarman opened on Jun 16, 2025

Parameter counting when quaxified layers are invoked

questionUser queriesUser queries

Status: Open.

#66 In patrick-kidger/quax;

· neel04 opened on Jun 12, 2025

Using quax with jax.jvp

Status: Open.

#65 In patrick-kidger/quax;

· nardi opened on Jun 9, 2025

Interest in benchmarking?

Status: Open.

#61 In patrick-kidger/quax;

· nstarman opened on Apr 23, 2025

Problem in <code>_Quaxify.__call__</code>

questionUser queriesUser queries

Status: Open.

#58 In patrick-kidger/quax;

· nstarman opened on Apr 14, 2025

Overloading <code>jax</code> operations to avoid undefined derivatives

questionUser queriesUser queries

Status: Open.

#35 In patrick-kidger/quax;

· alleSini99 opened on Nov 12, 2024

Class methods vs. quaxified functions

questionUser queriesUser queries

Status: Open.

#33 In patrick-kidger/quax;

· vadmbertr opened on Oct 15, 2024

<code>scan</code> primitive implementation missing

Status: Open.

#29 In patrick-kidger/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 patrick-kidger/quax;

· colehaus opened on Aug 28, 2024

jit and vmap best practices

documentationImprovements or additions to documentationImprovements or additions to documentation

Status: Open.

#11 In patrick-kidger/quax;

· nstarman opened on Feb 12, 2024