simonguo-cohere

simonguo-cohere's Stars

• HazyResearch/ThunderKittens

  Tile primitives for speedy kernels

  Language:Cuda1.5k58

• stanford-crfm/levanter

  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax

  Language:Python49378

• abhinavgoel95/collective-matmul-unit-tests

  Language:Python11

• google/paxml

  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.

  Language:Python44768

• NVIDIA/JAX-Toolbox

  JAX-Toolbox

  Language:Jupyter Notebook23144

• karpathy/llm.c

  LLM training in simple, raw C/CUDA

  Language:Cuda23.4k2.6k

• rwitten/HighPerfLLMs2024

  Language:Python20316