JR-1991 / CatalaxLinks
ποΈ - JAX-based framework to model biological systems
β15Updated last week
Alternatives and similar repositories for Catalax
Users that are interested in Catalax are comparing it to the libraries listed below
Sorting:
- jax library for E3 Equivariant Neural Networksβ216Updated last month
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.β108Updated 6 months ago
- Fully and Partially Bayesian Neural Netsβ75Updated 5 months ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.β39Updated 7 months ago
- Cryo electron microscopy image simulation and analysis built on JAX.β38Updated this week
- Gaussian Processes for Experimental Sciencesβ227Updated 3 months ago
- Implementation of Forward Laplacian algorithm in JAXβ86Updated last week
- Normalizing-flow enhanced sampling package for probabilistic inference in Jaxβ250Updated last month
- Stochastic Gradient MCMC for Jaxβ16Updated 4 months ago
- An implementation of SchNet in JAX and JAX-MD.β17Updated 3 years ago
- β189Updated last month
- Matrix-free linear algebra in JAX.β141Updated 2 weeks ago
- DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electrβ¦β59Updated 5 months ago
- Distance-based Analysis of DAta-manifolds in pythonβ134Updated 3 months ago
- Automatic Functional Differentiation in JAXβ77Updated 2 weeks ago
- Differentiate all the things!β157Updated last week
- A tool to optimize real world problemsβ107Updated last week
- A Library for Gaussian Processes in Chemistryβ240Updated 11 months ago
- E(n) Equivariant GNN in jaxβ14Updated 2 years ago
- Boltzmann Generators and Normalizing Flows in PyTorchβ175Updated last year
- β53Updated 2 years ago
- Interpolation and function approximation with JAXβ223Updated last week
- MESS: Modern Electronic Structure Simulationsβ38Updated last week
- An implementation of ESM2 in Equinox+JAXβ35Updated 4 months ago
- Equivariant machine learning interatomic potentials in JAX.β75Updated 5 months ago
- β32Updated 3 years ago
- Self-describing sparse tensor data format for atomistic machine learning and beyondβ84Updated this week
- β22Updated 5 months ago
- Differentiable scientific computing libraryβ155Updated last year
- Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.β69Updated last year