patrick-kidger/equinox

Issues

• Right way to count cache size

  #743 opened 11 days ago by AakashKumarNain
  2

• Test fail with latest jaxtyping (0.2.29)

  #740 opened 14 days ago by GaetanLepage
  2

• Error with data sharding

  #738 opened 13 days ago by andremfreitas
  5

• docs request for filter_jacs/hessian

  #741 opened 14 days ago by aidancrilly
  2

• A simple DeepMLP is taking 11GB/16GB in GPU RAM

  #737 opened 14 days ago by Qazalbash
  4

• Flash Attention

  #732 opened 22 days ago by neel04
  6

• Conv Transpose doesn't exactly transpose

  #728 opened 21 days ago by lockwo
  4

• Unroll in eqx's internal `scan`

  #726 opened a month ago by neel04
  11

• Kolmogorov Arnold Networks with equinox [kanx]

  #730 opened 23 days ago by stergiosba
  2

• Custom JVP/VJP definition within Module?

  #725 opened a month ago by leonard-gleyzer
  3

• Static/Dynamic issues with filter_eval_shape

  #729 opened 25 days ago by quattro
  1

• Jax 0.4.27: ValueError: safe_map() argument 2 is shorter than argument 1

  #716 opened a month ago by GaetanLepage
  7

• Possible memory leak?

  #673 opened a month ago by AakashKumarNain
  34

• Comparisons to NNX

  #718 opened a month ago by jlperla
  3

• Strange error with multiple tuples as attributes and `equinox.tree_at`?

  #715 opened a month ago by leonard-gleyzer
  5

• How to serialize a model + state pair?

  #714 opened a month ago by Artur-Galstyan
  2

• what does eqx.experimental.set_state do in the following code?

  #663 opened a month ago by Basant1861
  7

• merging multiple eqx.Module classes

  #713 opened a month ago by ToshiyukiBandai
  2

• Getting a type error when initializing Linear module (unexpected keyword argument 'key')

  #712 opened a month ago by AbhinavRao23
  1

• `eqx.tree_at` fails with `TypeError: Operation undefined, {x} is not a leaf of the pytree` when custom pytree attributes are modified within `__init__`

  #711 opened 2 months ago by nicolasJouvin
  5

• Weird `jax` error when trying vmap twice while using batchnorm

  #707 opened 2 months ago by PaulScemama
  5

• Regarding trainable parameters in equinox

  #710 opened 2 months ago by Basant1861
  9

• How to print HLO for jit accelated function?

  #700 opened 2 months ago by nanoric
  1

• lax.scan for equinox Modules ?

  #709 opened 2 months ago by mchagneux
  5

• Incorrect error handling in error_if

  #696 opened 2 months ago by danielward27
  3

• `UnexpectedTracerError` when vmapping `SpectralNorm`

  #705 opened 2 months ago by pablo2909
  6

• gradient with sequential and jax.nn.<activation functions>

  #706 opened 2 months ago by mfouesneau
  4

• Graph Neural Networks

  #702 opened 2 months ago by stergiosba
  1

• Broken link in documentation

  #698 opened 2 months ago by lockwo
  0

• Sharding - shard `eqx.Module` as well as inputs?

  #688 opened 2 months ago by homerjed
  19

• Add a [source] to docs [feature request]

  #683 opened 3 months ago by lockwo
  3

• Elegant way to index pytrees at arbitrary leaves using equinox?

  #695 opened 2 months ago by mjo22
  2

• Not using `filter_jit` produces `None` during backward pass

  #681 opened 3 months ago by neel04
  12

• Optimally training ensembles.

  #693 opened 3 months ago by BabaYara
  4

• Type annotations for "struct of arrays" ?

  #692 opened 3 months ago by nboyd
  1

• How to reduce compile time of loops over "PyTree Callables"?

  #685 opened 3 months ago by pfackeldey
  6

• Why is this function recompiling?

  #686 opened 3 months ago by jdpeiffer6
  2

• [Enhancement] Adding `dtype` to nn layers

  #679 opened 3 months ago by AakashKumarNain
  9

• Could filtered transformations accept an optional filter_spec?

  #682 opened 3 months ago by mjo22
  3

• What is the recommended pattern for optimization loops in transformed parameter spaces?

  #676 opened 3 months ago by mjo22
  9

• Feature request for custom behavior in apply_updates

  #678 opened 3 months ago by mjo22
  2

• Question: Support for Transformer Engine

  #669 opened 3 months ago by ConnorBaker
  4

• [Show and Tell] Jaxonloader - A dataloader for JAX built as eqx.Module

  #672 opened 3 months ago by Artur-Galstyan
  0

• Masking in Embedding layer

  #668 opened 3 months ago by AakashKumarNain
  1

• Recursively setting weights of submodules

  #657 opened 3 months ago by AakashKumarNain
  5

• BatchNorm training instability

  #659 opened 4 months ago by andrewdipper
  5

• eqx.nn.Conv2d slower than torch.nn.Conv2d

  #667 opened 4 months ago by lowlorenz
  1

• Speeding up Newton Update

  #666 opened 4 months ago by BabaYara
  2

• `nn.Linear` does not support broadcasting for the bias term

  #662 opened 4 months ago by ariG23498
  11

• Using an Equinox model outside Python - Deserialisation

  #664 opened 4 months ago by stergiosba
  1