richardk53

richardk53's Stars

• AI-Hypercomputer/maxtext

  A simple, performant and scalable Jax LLM!

  Language:Python1.5k292

• google/flax

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

  Language:Jupyter Notebook6.1k645

• NVIDIA/Megatron-LM

  Ongoing research training transformer models at scale

  Language:Python10.5k2.4k

• pytorch/torchtitan

  A native PyTorch Library for large model training

  Language:Python2.6k201

• mlcommons/inference

  Reference implementations of MLPerf™ inference benchmarks

  Language:Python1.2k533

• pandas-dev/pandas

  Flexible and powerful data analysis / manipulation library for Python, providing labeled data structures similar to R data.frame objects, statistical functions, and much more

  Language:Python43.7k17.9k

• EEA-sensors/parallel-non-linear-gaussian-smoothers

  Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.

  Language:Jupyter Notebook252

• HazyResearch/ThunderKittens

  Tile primitives for speedy kernels

  Language:Cuda1.6k66

• triton-lang/triton

  Development repository for the Triton language and compiler

  Language:C++13.3k1.6k

• jopetty/word-problem

  Experiments on the impact of depth in transformers and SSMs.

  Language:Python143

• state-spaces/mamba

  Mamba SSM architecture

  Language:Python13.1k1.1k

• sustcsonglin/flash-linear-attention

  Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton

  Language:Python1.3k68

• microsoft/Samba

  Official implementation of "Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling"

  Language:Python80148

• marimo-team/marimo

  A reactive notebook for Python — run reproducible experiments, execute as a script, deploy as an app, and version with git.

  Language:Python7.7k266

• HigherOrderCO/Bend

  A massively parallel, high-level programming language

  Language:Rust17.4k427

• alxndrTL/mamba.py

  A simple and efficient Mamba implementation in pure PyTorch and MLX.

  Language:Python1k92

• johnma2006/mamba-minimal

  Simple, minimal implementation of the Mamba SSM in one file of PyTorch.

  Language:Python2.6k191

• explosion/curated-transformers

  🤖 A PyTorch library of curated Transformer models and their composable components

  Language:Python86534

• vaexio/vaex

  Out-of-Core hybrid Apache Arrow/NumPy DataFrame for Python, ML, visualization and exploration of big tabular data at a billion rows per second 🚀

  Language:Python8.3k590

• google-deepmind/enn

  Language:Python29455

• google-deepmind/neural_testbed

  Language:Jupyter Notebook19116

• timeseriesAI/tsai

  Time series Timeseries Deep Learning Machine Learning Python Pytorch fastai | State-of-the-art Deep Learning library for Time Series and Sequences in Pytorch / fastai

  Language:Jupyter Notebook5.2k656

• mbilos/neural-flows-experiments

  Experiments for Neural Flows paper

  Language:Python8517

• poets-ai/elegy

  A High Level API for Deep Learning in JAX

  Language:Python46932

• dtcooper/raspotify

  A Spotify Connect client that mostly Just Works™

  Language:Shell4.7k217

• Lightning-AI/pytorch-lightning

  Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

  Language:Python28.3k3.4k

• awslabs/gluonts

  Probabilistic time series modeling in Python

  Language:Python4.6k753

• juliuskunze/jaxnet

  Concise deep learning for JAX

  Language:Python18414

• SPFlow/SPFlow

  Sum Product Flow: An Easy and Extensible Library for Sum-Product Networks

  Language:Python28981

• minosworld/minos

  MINOS: Multimodal Indoor Simulator

  Language:Python20133