nvidia-cosmos / cosmos-xennaLinks
Python library for building and running distributed data pipelines using Ray
☆37Updated 3 weeks ago
Alternatives and similar repositories for cosmos-xenna
Users that are interested in cosmos-xenna are comparing it to the libraries listed below
Sorting:
- CATransformers is a framework for joint neural network and hardware architecture search.☆17Updated 5 months ago
- Cosmos-RL is a flexible and scalable Reinforcement Learning framework specialized for Physical AI applications.☆190Updated this week
- Cosmos-Curate is a powerful video curation system that processes, analyzes, and organizes video content using advanced AI models and dist…☆87Updated 2 weeks ago
- Multi-Turn RL Training System with AgentTrainer for Language Model Game Reinforcement Learning☆48Updated this week
- MobileLLM-R1☆44Updated 2 weeks ago
- [WIP] Better (FP8) attention for Hopper☆33Updated 7 months ago
- ☆46Updated last year
- RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best…☆53Updated 7 months ago
- Implementation of SmoothCache, a project aimed at speeding-up Diffusion Transformer (DiT) based GenAI models with error-guided caching.☆45Updated 3 months ago
- Train, tune, and infer Bamba model☆134Updated 4 months ago
- imagetokenizer is a python package, helps you encoder visuals and generate visuals token ids from codebook, supports both image and video…☆37Updated last year
- Official Code Repository for EnvGen: Generating and Adapting Environments via LLMs for Training Embodied Agents (COLM 2024)☆38Updated last year
- Pixel Parsing. A reproduction of OCR-free end-to-end document understanding models with open data☆21Updated last year
- Multimodal Open Source Framework for Conversational Agent Research and Development.☆21Updated 8 months ago
- Implementation of a methodology that allows all sorts of user defined GPU kernel fusion, for non CUDA programmers.☆25Updated this week
- 😊 TPTT: Transforming Pretrained Transformers into Titans☆29Updated 3 weeks ago
- CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learning☆192Updated 2 months ago
- DPO, but faster 🚀☆45Updated 10 months ago
- Pytorch DTensor native training library for LLMs/VLMs with OOTB Hugging Face support☆110Updated this week
- Experimental scripts for researching data adaptive learning rate scheduling.☆22Updated 2 years ago
- Residual Quantization with Implicit Neural Codebooks☆103Updated last week
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆130Updated 10 months ago
- Training-free Post-training Efficient Sub-quadratic Complexity Attention. Implemented with OpenAI Triton.☆147Updated this week
- Compression for Foundation Models☆35Updated 2 months ago
- ☆88Updated last year
- vLLM adapter for a TGIS-compatible gRPC server.☆41Updated this week
- Just another reasonably minimal repo for class-conditional training of pixel-space diffusion transformers.☆130Updated 4 months ago
- Multi-Layer Key-Value sharing experiments on Pythia models☆34Updated last year
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆114Updated 2 months ago
- A collection of reproducible inference engine benchmarks☆34Updated 5 months ago