JR-1991 / CatalaxLinks
ποΈ - JAX-based framework to model biological systems
β15Updated 2 weeks ago
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 2 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.β109Updated 6 months ago
- Fully and Partially Bayesian Neural Netsβ75Updated 5 months ago
- Implementation of Forward Laplacian algorithm in JAXβ88Updated last month
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.β39Updated 7 months 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β¦β62Updated 6 months ago
- Self-describing sparse tensor data format for atomistic machine learning and beyondβ87Updated 2 weeks ago
- β53Updated 3 years ago
- An implementation of SchNet in JAX and JAX-MD.β17Updated 3 years ago
- Gaussian Processes for Experimental Sciencesβ228Updated 3 months ago
- tools for graph-based machine-learning potentials in jaxβ25Updated last year
- Equivariant machine learning interatomic potentials in JAX.β76Updated last week
- Particle-mesh based calculations of long-range interactions in JAXβ19Updated this week
- Cryo electron microscopy image simulation and analysis built on JAX.β38Updated this week
- A flexible and performant framework for training machine learning potentials.β30Updated this week
- MESS: Modern Electronic Structure Simulationsβ39Updated last month
- Differentiate all the things!β158Updated last month
- Stochastic Gradient MCMC for Jaxβ16Updated 5 months ago
- Distance-based Analysis of DAta-manifolds in pythonβ134Updated 3 weeks ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-fβ¦β31Updated last year
- Tools for building equivariant polynomials on reductive Lie groups.β35Updated 2 years ago
- GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine lβ¦β107Updated last year
- Differentiable scientific computing libraryβ155Updated last year
- Modeling toolkit for biochemical simulationβ79Updated 9 months ago
- A simple Python wrapper for Slurm with flexibility in mind.β158Updated 5 months ago
- β22Updated 5 months ago
- Materials Learning Algorithms. A framework for machine learning materials properties from first-principles data.β97Updated last month
- Pytorch differentiable molecular dynamicsβ179Updated 3 years ago
- Exchange correlation functionals translated from libxc to jaxβ46Updated 7 months ago
- Matrix-free linear algebra in JAX.β141Updated last month