Shin

tpux. A set of Python scripts that makes your experience on TPU better

mistral-v0.2-jax. JAX implementation of the Mistral 7b v0.2 model

einshard. Einsum-like high-level array sharding API for JAX

tpu-training-example. Python

mistral-jax. JAX implementation of the Mistral 7b v0.1 model

schedule-main. Terminal-based event tracking tool.

MahjongGenerator. A customizable Mahjong game

telegram-bots. A collection of my telegram bots

bark-jax. JAX implementation of Bark

learn-jax-with-code. A collection of code to help understand features in JAX

encodec-jax. JAX implementation of EnCodec

optax. Optax is a gradient processing and optimization library for JAX.

go. The Go programming language

ColossalAI. Making large AI models cheaper, faster and more accessible

triton. Development repository for the Triton language and compiler

Shin-website. Personal Website: https://yixiaoer.sg/

transformers. 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

flax. Flax is a neural network library for JAX that is designed for flexibility.

girlspowertech.github.io. Home Page

jax. Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

gemma. Open weights LLM from Google DeepMind.

rust. Empowering everyone to build reliable and efficient software.

linux. Linux kernel source tree

flash-attention. Fast and memory-efficient exact attention

orbax. Orbax provides common utility libraries for JAX users.

cloud-accelerator-diagnostics. Python

Bingyan-practice. Go

llama2.c. Inference Llama 2 in one file of pure C

dstack. dstack is an easy-to-use and flexible container orchestrator for running AI workloads in any cloud or data center.

dive. A tool for exploring each layer in a docker image