Code for the paper Normalizing Flows are Capable Models for RL

code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"

27-Queens Puzzle: Massively Parellel Enumeration and Solution Counting

Library for Crystal Symmetry in Rust

Different utilities used by our group

Quantum error correction code AI-discovery with Jax

Tools to Build Equivariant Tensors

Torch-native, batchable, atomistic simulation.

Foundation Model for Materials - FM4M

PyTorch implementation of FractalGen https://arxiv.org/abs/2502.17437

Model Context Protocol Servers

Repository for material accompanying the ptvmc systematic study paper

A Curated Dataset of Crystal Structures and Experimentally Measured Ionic Conductivities for Lithium Solid-State Electrolytes

Conscious alternative to LinearAlgebra.jl

Hex board game AI with self-play learning based on the AlphaZero algorithm

OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

Gaussianization Flows for density estimation, sampling and information theory measures (RBIG2.0)

Hierarchical Tucker for Black Box approximation and optimization

A PyTorch implementation of TTGO algorithm and the applications presented in the paper "Tensor Train for Global Optimization Problems in Robotics"

Integer Programming encoding for Crystal Structure Prediction with classic and quantum computing bindings

CrystalGRW: Generative Modeling of Crystal Structures with Targeted Properties via Geodesic Random Walks

Python implementation of the TT-Toolbox

Library for computing with multidimensional functions

Tensor-Train decomposition in pytorch

Sampling and visualisation of the von Mises distribution