patrick-kidger equinox issues

patrick-kidger / equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Apache License 2.0

2.13k stars 143 forks source link

issues

Newest

Newest Most commented Recently updated Oldest Least commented Least recently updated

eqx.filter_hessian gives incorrect result for jax.nn.relu

#801 SimonKoop closed 2 weeks ago
1
Disable static arrays

#800 lockwo closed 3 months ago
17
Added the offset to RoPE embedding and fixed the pre-commit pyright

#799 Artur-Galstyan closed 2 months ago
6
Realistic scenario in which jax array is static

#798 lockwo closed 3 months ago
4
Freezing layers with `eqx.tree_map` (not parameters in a single layer)

#797 homerjed closed 3 months ago
2
[Question] Interaction with optax LBFGS optimizer

#796 bnesposito opened 3 months ago
3
Fix missing `out_axes` for transpose-of-vprim.

#795 patrick-kidger closed 3 months ago
0
Optax with multiple optimizers

#794 ToddMorrill opened 3 months ago
3
[Question] Best practices for profiling Equinox models

#793 AakashKumarNain opened 3 months ago
9
Equivalent of flax's TrainState

#792 misterguick closed 3 months ago
2
Calling eqx.Module method invokes user defined __getattr__

#791 yuzu-ido closed 3 months ago
2
Best way to do complex weight sharing

#790 neel04 opened 4 months ago
4
Filter jax lax primitives

#789 DrJessop closed 4 months ago
1
better bug hint when writing a simple neural network in equinox

#788 zhengqigao opened 4 months ago
1
Does equinox have an example or opensource model for distributed training on multi-hosts

#787 knightXun closed 4 months ago
0
Ensemble model training/running with different architectures

#786 PeishiJiang closed 4 months ago
6
Equinox runtime errors now include a stack trace and more debug info.

#785 patrick-kidger closed 4 months ago
0
best way to write a wrapper function for tree_at

#784 sede-fa opened 4 months ago
1
Better strict error

#783 lockwo closed 4 months ago
1
Better error messages for strict

#782 lockwo closed 4 months ago
1
test_traceback_runtime_eqx fails with latest jaxtyping (0.2.33)

#781 GaetanLepage closed 2 months ago
3
MLP "depth" parameter doc

#780 tyagi-ankit opened 4 months ago
2
Conv2dTranspose possible bug??

#779 haroldle closed 4 months ago
4
Initialization of large models on multi-hosts environment

#778 kazewong opened 4 months ago
9
Fixes

#777 patrick-kidger closed 4 months ago
0
Vmap and lax.scan with different sequence array

#776 prajwalthakur closed 3 months ago
2
Bug: typing issue due to `__getattribute__`

#775 nstarman opened 4 months ago
4
Increase minimum required python version to 3.10

#774 Artur-Galstyan opened 4 months ago
2
How to extract learnable parameters of a subnetwork

#773 itk22 closed 4 months ago
2
Dropout causes `None`s with gradient accumulation

#772 neel04 opened 5 months ago
8
Recommended way of filtering params for weight decay

#771 AakashKumarNain closed 4 months ago
20
`jit`ing large models for inference has bad compilation performance

#770 colehaus opened 5 months ago
3
Fix for a case where we're hashing tracers.

#769 patrick-kidger closed 5 months ago
4
_TrivialClosureConvert mis-specifies dynamic attributes

#768 jakevdp closed 5 months ago
3
Removed use of deprecated jax.core.pp_*

#767 patrick-kidger closed 5 months ago
0
Remove references to internal `jax.core.pp_*` functions

#766 jakevdp closed 5 months ago
2
Support complex dtypes in networks

#765 ChenAo-Phys closed 5 months ago
3
Export attention function similar to torch.nn.functional.scaled_dot_product_attention

#764 Artur-Galstyan opened 5 months ago
1
`Conv` doesn't support complex dtype

#763 ChenAo-Phys closed 5 months ago
2
`warn_if`

#762 nstarman opened 5 months ago
3
Add a `warn_if`

#761 nstarman opened 5 months ago
1
Given a state, how to get the values of its arrays?

#760 Artur-Galstyan closed 5 months ago
1
Question: `filter_shard` with PartitionSpecs, or other ways to ensure batched output, as in `vmap`.

#759 johannahaffner opened 5 months ago
3
Fix more JAX deprecation warnings.

#758 patrick-kidger closed 5 months ago
0
Broken links and how to find them

#757 matteoguarrera closed 5 months ago
2
Question - How to assert that a `filter_spec` is valid?

#756 itk22 closed 5 months ago
2
`filter_shard` test crashes on JAX 0.4.29

#755 patrick-kidger opened 5 months ago
4
Doc typo!

#754 patrick-kidger closed 5 months ago
0
Update users of jax.tree.map() to be more careful about how they hand…

#753 hawkinsp closed 5 months ago
1
Fixes edge-case crash with CWL

#752 patrick-kidger closed 5 months ago
0

Previous Next