Rigorous Proofs via Adversarial AI Agents

Implementation for the MP-NQS

Fast low-rank updates (LRU) of matrix determinants and pfaffians in JAX

TREX I/O library

BFD pseudopotentials and basis sets

A repository for quantum chemistry basis sets

Operate and manipulate physical quantities in Python

MLX-VLM is a package for inference and fine-tuning of Vision Language Models (VLMs) on your Mac using MLX.

Reusable array functionality for Julia's various GPU backends.

A fast stabilizer circuit library.

An Extensible Deep Learning Library

Kolmogorov Arnold Networks

Accompanying code for "Accurate neural quantum states for interacting lattice bosons"

User-friendly exact diagonalization package written in Haskell. Can treat systems of up to 𝒪(42) spins!

nanobind: tiny and efficient C++/Python bindings

Differentiable, Hardware Accelerated, Molecular Dynamics

Neural Networks for JAX

OpTree: Optimized PyTree Utilities

Unbearably fast near-real-time pure-Python runtime-static type-checker.

Implementation of Nougat Neural Optical Understanding for Academic Documents

An llvm-based framework for generating and calling into high-performance native code from Python.

Libint: high-performance library for computing Gaussian integrals in quantum mechanics

A dead simple Python package for creating custom JAX pytree objects

Orbax provides common checkpointing and persistence utilities for JAX users

A Graph Neural Network Library in Jax

Python compiler

A machine learning compiler for GPUs, CPUs, and ML accelerators

GPS plugin for NetKet (www.netket.org), introducing new models, optimizers and Fermionic functionality.

A high-performance, zero-overhead, extensible Python compiler with built-in NumPy support