issues
search
e3nn
/
e3nn-jax
jax library for E3 Equivariant Neural Networks
Apache License 2.0
182
stars
18
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
jax v0.4.34 breaks e3nn_jax import
#83
BoltMaxwell
opened
4 weeks ago
2
optimzer state during e3nn model training
#81
VinaySingh561
closed
1 month ago
1
Enquiring About `e3nn_jax` Support for the New `flax.nnx` Framework
#80
kYangLi
opened
1 month ago
2
Fix documentation build errors.
#79
ameya98
closed
2 months ago
0
Add custom initializers and instructions in Linear.
#78
ameya98
closed
2 months ago
0
Add custom initialization options in e3nn_jax.flax.Linear
#77
chen-zz20
closed
2 months ago
7
Updated citation
#76
mitkotak
closed
3 months ago
0
Cache cg coefficients 2
#73
mitkotak
closed
4 months ago
1
Please wrap `clebsch_gordan` inside `functools.cache`
#72
olivier-peltre
closed
5 months ago
1
Cache CG coefficients
#71
olivier-peltre
closed
4 months ago
14
IrrepsArray.transform_by_angles and LinearSHTP does not support batch operation
#70
AdaaaWang
closed
5 months ago
2
"tensor_product_with_spherical_harmonics" slower + uses more memory
#69
Kalyan0821
closed
6 months ago
8
TensorProduct and FullyConnectedTensorProduct in e3nn_jax
#67
VinaySingh561
closed
7 months ago
1
Code coverage not working
#65
mariogeiger
closed
7 months ago
9
Add ones() and ones_like() with tests.
#64
ameya98
closed
10 months ago
1
Legendre Transform tests are failing
#63
ameya98
closed
10 months ago
1
Add Linear layer in Equinox with all tests passing!
#62
ameya98
closed
10 months ago
1
GitHub install fails silently
#61
ameya98
closed
10 months ago
3
Fix python 3.9 compatibility
#60
pabloferz
closed
10 months ago
2
Add special case for lmax=0 to avoid NaN.
#59
ameya98
closed
10 months ago
4
feat(ci): enable pip cache + restructure
#58
SauravMaheshkar
closed
10 months ago
0
chore: move BENCHMARK to docs
#57
SauravMaheshkar
closed
10 months ago
2
refactor test suite
#56
SauravMaheshkar
closed
10 months ago
0
refactor test suite
#55
SauravMaheshkar
closed
10 months ago
1
feat: migrate packaging to `pyproject.toml`
#54
SauravMaheshkar
closed
10 months ago
1
migrate to `pyproject.toml` for packaging
#53
SauravMaheshkar
closed
10 months ago
1
docs: fix link for CHANGELOG
#52
SauravMaheshkar
closed
10 months ago
0
chore: migrate style config to `pyproject.toml`
#51
SauravMaheshkar
closed
10 months ago
0
Fix resolution check.
#49
ameya98
closed
11 months ago
0
S2grid
#48
ameya98
closed
11 months ago
0
Add S2FFT and other SphericalSignal functionality
#47
ameya98
closed
11 months ago
0
add fused spherical harmonics
#46
bhedelius
closed
9 months ago
4
FullyConnectedTensorProduct Feature Discrepancy in JAX vs. Torch
#44
tai-dang11
closed
1 year ago
2
Availability of FullTensorProduct function similar to PyTorch version
#43
tai-dang11
closed
1 year ago
2
export legendre helper functions
#42
songk42
closed
1 year ago
2
multiple Legendre transforms
#41
songk42
closed
1 year ago
0
Caching CG coeffs
#40
mitkotak
closed
1 year ago
0
[WIP] Add Equinox implementation of Linear.
#39
ameya98
closed
10 months ago
4
Backward pass runtime degradation (Linear + `tensor_product`) in the latest versions
#38
gerkone
closed
11 months ago
8
Legendre transform
#37
songk42
closed
1 year ago
7
Move complicated implementation of TP into Legacy submodule
#36
mariogeiger
closed
1 year ago
0
IrrepsArray more jax pytree and jax transform safe
#35
mariogeiger
closed
1 year ago
0
Make IrrepsArray frozen
#34
mariogeiger
closed
1 year ago
1
Utils
#33
mariogeiger
closed
1 year ago
0
optimize reduced_symmetric_tensor_product_basis
#32
mariogeiger
closed
1 year ago
0
Fix radius option for scale_radius_by_amplitude.
#26
ameya98
closed
1 year ago
1
Adds final tests for SphericalSignal.
#25
ameya98
closed
1 year ago
0
Add SphericalSignal for manipulating signals on S2.
#24
ameya98
closed
1 year ago
3
Fix dtypes as input.array.dtype for tests to pass.
#23
ameya98
closed
1 year ago
0
Add optimized_reduced_symmetric_tensor_product().
#22
ameya98
closed
1 year ago
0
Next