AaltoSciComp / hpc-examplesLinks
Examples of Triton usage
☆22Updated 3 months ago
Alternatives and similar repositories for hpc-examples
Users that are interested in hpc-examples are comparing it to the libraries listed below
Sorting:
- Exchange correlation functionals translated from libxc to jax☆45Updated 2 months ago
- Solving Optimization Problems with JAX, code and PDF☆16Updated 5 years ago
- This is a lightweight python script to abbreviate the journal names in your latex bib file.☆45Updated last year
- A software package for flexible HPC GPs☆16Updated last week
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Automatic Functional Differentiation in JAX☆72Updated 2 months ago
- ☆19Updated 2 years ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 3 months ago
- Matrix-free linear algebra in JAX.☆121Updated last week
- Python Tensor Toolbox☆33Updated last month
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Updated 2 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Python Algorithms for Randomized Linear Algebra☆54Updated 2 years ago
- ☆20Updated 2 years ago
- Flexible, general-purpose VMC framework, built on JAX.☆26Updated last month
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆111Updated 2 months ago
- A generic interface for linear algebra backends☆73Updated 3 months ago
- Hierarchical variational models for physics.☆17Updated 5 years ago
- JAX accelerated Quantum Monte Carlo☆78Updated last week
- Tensor Train Toolbox☆113Updated last month
- ☆175Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆128Updated 8 months ago
- Library for normalizing flows and neural flows.☆24Updated 3 years ago
- jax library for E3 Equivariant Neural Networks☆207Updated 4 months ago
- Code for efficiently sampling functions from GP(flow) posteriors☆72Updated 4 years ago
- Materials Learning Algorithms. A framework for machine learning materials properties from first-principles data.☆90Updated this week
- Multi-GPU training with TensorFlow on Piz Daint☆12Updated 3 years ago
- Fully and Partially Bayesian Neural Nets☆70Updated last month
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆277Updated this week