nvidia-cosmos / cosmos-xennaLinks
Python library for building and running distributed data pipelines using Ray
☆44Updated last week
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 6 months ago
- Multi-Turn RL Training System with AgentTrainer for Language Model Game Reinforcement Learning☆48Updated last week
- Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support☆147Updated this week
- Cosmos-RL is a flexible and scalable Reinforcement Learning framework specialized for Physical AI applications.☆204Updated last week
- Cosmos-Curate is a powerful video curation system that processes, analyzes, and organizes video content using advanced AI models and dist…☆101Updated last week
- Pixel Parsing. A reproduction of OCR-free end-to-end document understanding models with open data☆21Updated last year
- ☆46Updated last year
- DPO, but faster 🚀☆46Updated 11 months ago
- Experimental scripts for researching data adaptive learning rate scheduling.☆22Updated 2 years ago
- Train, tune, and infer Bamba model☆135Updated 5 months ago
- 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
- ☆55Updated 4 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆130Updated 11 months ago
- 😊 TPTT: Transforming Pretrained Transformers into Titans☆29Updated 3 weeks ago
- Description and applications of OpenAI's paper about DALL-E (2021) and implementation of other (CLIP-guided) zero-shot text-to-image gene…☆32Updated 3 years ago
- CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learning☆203Updated last week
- ☆90Updated last year
- imagetokenizer is a python package, helps you encoder visuals and generate visuals token ids from codebook, supports both image and video…☆37Updated last year
- MobileLLM-R1☆55Updated last month
- Compression for Foundation Models☆35Updated 3 months ago
- QeRL enables RL for 32B LLMs on a single H100 GPU.☆416Updated 3 weeks ago
- Easily run PyTorch on multiple GPUs & machines☆47Updated last week
- Implementation of BitNet-1.58 instruct tuning☆27Updated last year
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆104Updated last year
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆40Updated last year
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆36Updated last year
- Official Code Repository for EnvGen: Generating and Adapting Environments via LLMs for Training Embodied Agents (COLM 2024)☆38Updated last year
- LL3M: Large Language and Multi-Modal Model in Jax☆74Updated last year
- ☆26Updated last month
- Code for “Pretrained Language Models as Visual Planners for Human Assistance”☆61Updated 2 years ago