Stars
Xwin-LM: Powerful, Stable, and Reproducible LLM Alignment
Single-line inference of SOTA deep learning models
jax-triton contains integrations between JAX and OpenAI Triton
Infer.NET is a framework for running Bayesian inference in graphical models
YTsaurus is a scalable and fault-tolerant open-source big data platform.
🦜🔗 Build context-aware reasoning applications
Evals is a framework for evaluating LLMs and LLM systems, and an open-source registry of benchmarks.
Orbax provides common checkpointing and persistence utilities for JAX users
Nethack Learning Environment Wrapper for Language Interface
A toolkit for developing and comparing reinforcement learning algorithms.
Development repository for the Triton language and compiler
Flax is a neural network library for JAX that is designed for flexibility.
High-quality implementations of standard and SOTA methods on a variety of tasks.
A concise but complete full-attention transformer with a set of promising experimental features from various papers
Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure as a dataclass and let pyrallis do the rest!
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
YDB is an open source Distributed SQL Database that combines high availability and scalability with strong consistency and ACID transactions