#
Lists (32)
Build
6 repositories
Computer Vision
3 repositories
Cookbooks
9 repositories
custom-trainer
13 repositories
Dagster
1 repository
Data-infra
5 repositories
Docs
2 repositories
Finance
10 repositories
Information Retrieval
10 repositories
Jax
14 repositories
Large Language Models
39 repositories
Large scale ML
3 repositories
LLM Eval
1 repository
LLM Rapids
3 repositories
LLM + Tabular
5 repositories
MCP
3 repositories
meshx
4 repositories
ML
7 repositories
ML Executor
17 repositories
ML-Infra
14 repositories
NeMo
11 repositories
NeMo Agent
15 repositories
PKM
78 repositories
Python
44 repositories
Pytorch
15 repositories
Recsys
26 repositories
RL
1 repository
Rust
14 repositories
Scripts
1 repository
Shell
1 repository
Tensorflow
3 repositories
Vscode
3 repositories
Starred repositories
727
stars
written in Python
Clear filter
A logical, reasonably standardized, but flexible project structure for doing and sharing data science work.
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, 17+ clouds, or on-prem).
Out-of-Core hybrid Apache Arrow/NumPy DataFrame for Python, ML, visualization and exploration of big tabular data at a billion rows per second 🚀
Ready-to-run Docker images containing Jupyter applications
An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.
The property-based testing library for Python
tensorboard for pytorch (and chainer, mxnet, numpy, ...)
Easy-to-use,Modular and Extendible package of deep-learning based CTR models .
Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM
High accuracy RAG for answering questions from scientific documents with citations
DoWhy is a Python library for causal inference that supports explicit modeling and testing of causal assumptions. DoWhy is based on a unified language for causal inference, combining causal graphic…
The official repo for “Dolphin: Document Image Parsing via Heterogeneous Anchor Prompting”, ACL, 2025.
A debugging and profiling tool that can trace and visualize python code execution
Tools for merging pretrained large language models.
A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility
A lightweight, object-oriented finite state machine implementation in Python with many extensions
Setup and customize deep learning environment in seconds.
Kats, a kit to analyze time series data, a lightweight, easy-to-use, generalizable, and extendable framework to perform time series analysis, from understanding the key statistics and characteristi…
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.
Pydoll is a library for automating chromium-based browsers without a WebDriver, offering realistic interactions.
Adding guardrails to large language models.