pierreglaser

llm.c

LLM training in simple, raw C/CUDA

alpaca.cpp

Locally run an Instruction-Tuned Chat-Style LLM

pixi

Package management made easy

emcee

The Python ensemble sampling toolkit for affine-invariant MCMC

graphein

Protein Graph Library

onnx-mlir

Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure

daft

Render probabilistic graphical models using matplotlib

dynamax

State Space Models library in JAX

anki-manual

Anki's manual

pdb-tools

A dependency-free cross-platform swiss army knife for PDB files.

extending-jax

Extending JAX with custom C++ and CUDA code

AllenSDK

code for reading and processing Allen Institute for Brain Science data

grayskull

Grayskull :skull: - Recipe generator for Conda

nanodl

A Jax-based library for designing and training transformer models from scratch.

neuromaps

A toolbox for comparing brain maps

rwkvstic

Framework agnostic python runtime for RWKV models

bayeux

State of the art inference for your bayesian models.

neuro4ml.github.io

Neuroscience for machine learners course

MicroCanonicalHMC

MCHMC: sampler from an arbitrary differentiable distribution

mouse_connectivity_models

Python package providing mesoscale connectivity models for mouse.

pytorch2jax

Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.

bazinet_assortativity

Code supporting Bazinet et al., 2023, "Assortative mixing in micro-architecturally annotated brain connectomes" (Nature Communications))

mmdfuse

MMD-FUSE package implementing the MMD-FUSE test proposed in MMD-FUSE: Learning and Combining Kernels for Two-Sample Testing Without Data Splitting by Biggs, Schrab, and Gretton: https://arxiv.org/abs/2306.08777

agginc

AggInc package implementing the MMDAggInc, HSICAggInc and KSDAggInc tests proposed in Efficient Aggregated Kernel Tests using Incomplete U-statistics by Schrab, Kim, Guedj and Gretton: https://arxiv.org/abs/2206.09194 NeurIPS 2022

gen_models_dont_go_mad

Code to reproduce the experiments of the ICLR 2024 (spotlight) paper "On the Stability of Iterative Retraining of Generative Models on their Own Data" https://arxiv.org/abs/2310.00429

DeepLearningTutorial

A colab-style DL tutorial for how to build and train DNNs (with and without autograd packages) from scratch.

ThetaSequencesAreEligibilityTraces

Code for my paper: "Theta sequences as eligibility traces: a biological solution to credit assignment"

bib_cleaner

kccsd

slurm_jax_multinode