c0nn3r

vllm

A high-throughput and memory-efficient inference and serving engine for LLMs

blackboard

A (very) unfinished OpenType Math layout and rendering engine in Racket

MatX

An efficient C++17 GPU numerical computing library with Python-like syntax

prometeo

An experimental Python-to-C transpiler and domain specific language for embedded high-performance computing

s4

Structured state space sequence models

pytorch_scatter

PyTorch Extension Library of Optimized Scatter Operations

sparse_continuous_distributions

This repository provides open-source code for sparse continuous distributions and corresponding Fenchel-Young losses.

equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

cascadia-code

This is a fun, new monospaced font that includes programming ligatures and is designed to enhance the modern look and feel of the Windows Terminal.

dm_pix

PIX is an image processing library in JAX, for JAX.

hash-prospector

Automated integer hash function discovery

alphafold

Open source code for AlphaFold.

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

reticle

seqio

Task-based datasets, preprocessing, and evaluation for sequence models.

gtn

Automatic differentiation with weighted finite-state transducers.

ckconv

Code repository of the paper "CKConv: Continuous Kernel Convolution For Sequential Data" published at ICLR 2022. https://arxiv.org/abs/2102.02611

sdf

Simple SDF mesh generation in Python

cunumeric

An Aspiring Drop-In Replacement for NumPy at Scale

storchastic

Stochastic Automatic Differentiation library for PyTorch.

tensor_annotations

Annotating tensor shapes using Python types

egg

egg is a flexible, high-performance e-graph library

Metatheory.jl

General purpose algebraic metaprogramming and symbolic computation library for the Julia programming language: E-Graphs & equality saturation, term rewriting and more.

FastTransforms

:bullettrain_front: Fast orthogonal polynomial transforms :surfer:

Symbolics.jl

Symbolic programming for the next generation of numerical software

online_gp

Code repo for "Kernel Interpolation for Scalable Online Gaussian Processes"

NeuralCDE

Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

signatory

Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)

tensorized-rnn

A fully tensorized recurrent neural network using tensor-train decomposition

hippo-code