Tanay Mehta's repositories
mamba-train
A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM
gpt-search
Use GPT-3 and GPT-3.5 to query your documents!
gpt-pre-training-scratch
Basic GPT trained on Harry Potter books
fim-pretraining
Fill-in-the-Middle Pre-training of Large Language Models in pure PyTorch!
depression-sentiment-prediction
Depression Sentiment Prediction Project
flax-vision
Computer Vision Models from PyTorch, Tensorflow now for Flax!
GitHubGraduation-2022
Join the GitHub Graduation Yearbook and "walk the stage" on June 11.
implementations
This repository houses my implementations of different papers and interesting architectures.
lancedb-haystack
An integration of LanceDB vector database backend with Haystack
machine-learning-interview
Machine Learning Interviews from FAANG, Snapchat, LinkedIn. I have offers from Snapchat, Coupang, Stitchfix etc. Blog: mlengineer.io.
mamba-chat
Mamba-Chat: A chat LLM based on the state-space model architecture 🐍
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
composer
Train neural networks up to 7x faster
cuml
cuML - RAPIDS Machine Learning Library
elegy
A High Level API for Deep Learning in JAX
flappy-bird-gym
An OpenAI Gym environment for the Flappy Bird game
introduction
Introduction repository where you will be practicing basic Git skills
lance
Modern columnar data format for ML and LLMs implemented in Rust. Convert from parquet in 2 lines of code for 100x faster random access, vector index, and data versioning. Compatible with Pandas, DuckDB, Polars, Pyarrow, with more integrations coming..
lance-deeplearning-recipes
Deep Learning how-to's using Lance file format
latex-css
LaTeX.css is a CSS library that makes your website look like a LaTeX document
llama3
The official Meta Llama 3 GitHub site
OLMo
Modeling, training, eval, and inference code for OLMo, now using Lance for faster data I/O
optax
Optax is a gradient processing and optimization library for JAX.
rl_project
RL Project Repo
simplicity
Lightweight open-source perplexity
torch-lance
A standalone utility to save a PyTorch model in Lance file format
vectordb-recipes
High quality resources & applications for LLMs, multi-modal models and VectorDBs