Yan Ru Pei

@PeaBrane · User

GitHub profile ↗ · Compare

AI, physics, and vibes | Dynamo @ NVIDIA

NVIDIAIrvine74 followers39 repositories

Repositories

PeaBrane/nishimori-boundaries

Computer-assisted certificates for vertical and reentrant ferromagnetic boundaries below the Nishimori point

★ 0PythonForks 0

PeaBrane/mamba-tiny

Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).

★ 135PythonForks 15

PeaBrane/rbm_jax

A minimal example for training an RBM using jax.

★ 2Jupyter NotebookForks 0

PeaBrane/cutile-python

cuTile is a programming model for writing parallel kernels for NVIDIA GPUs

★ 0PythonForks 0

PeaBrane/sglang

SGLang is a fast serving framework for large language models and vision language models.

★ 0Forks 0

PeaBrane/torch-pesq

PyTorch implementation of the Perceptual Evaluation of Speech Quality for wideband audio

★ 1PythonForks 1

PeaBrane/dynamo

A Datacenter Scale Distributed Inference Serving Framework

★ 0Forks 0

PeaBrane/vllm

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

★ 0PythonForks 0

PeaBrane/FastSpeech2

An implementation of Microsoft's "FastSpeech 2: Fast and High-Quality End-to-End Text to Speech"

★ 0Forks 0