Jamba15 / SpectralToolsLinks
Spectral analysis and training of dense layers
☆17Updated last year
Alternatives and similar repositories for SpectralTools
Users that are interested in SpectralTools are comparing it to the libraries listed below
Sorting:
- Exact OU processes with JAX☆49Updated 4 months ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- Graph neural networks in JAX.☆67Updated last year
- Distance-based Analysis of DAta-manifolds in python☆131Updated 3 weeks ago
- A paper describing the implementation of PySR and SymbolicRegression.jl☆58Updated last year
- ☆51Updated 2 years ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆131Updated last year
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- DISCS: The code base for the Benchmark for Discrete Sampling☆57Updated 11 months ago
- Implementation of approximate free-energy minimization in PyTorch☆19Updated 3 years ago
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆80Updated 5 years ago
- Message Passing Neural Networks for Simplicial and Cell Complexes☆164Updated 2 years ago
- SEML: Slurm Experiment Management Library☆181Updated 2 months ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆80Updated last year
- A collection of graph neural networks implementations in JAX☆33Updated last year
- Hierarchical Associative Memory User Experience☆101Updated last week
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆30Updated 2 years ago
- COMP760 Lecture Notes☆26Updated 2 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆281Updated last week
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆56Updated 2 years ago
- Running Jax in PyTorch Lightning☆106Updated 7 months ago
- jax library for E3 Equivariant Neural Networks☆211Updated 6 months ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆84Updated last year
- Multiple dispatch over abstract array types in JAX.☆126Updated last month
- Generative Flow Networks - GFlowNet☆259Updated last week
- About A collection of AWESOME things about information geometry Topics☆164Updated last year
- Turning SymPy expressions into PyTorch modules.☆148Updated 2 years ago
- Riemannian Optimization Using JAX☆50Updated last year
- Library for normalizing flows and neural flows.☆24Updated 3 years ago
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆27Updated last month