artemZholus

artemZholus's Stars

• tinygrad/tinygrad

  You like pytorch? You like micrograd? You love tinygrad! ❤️

  Language:Python27.3k2748343k

• pybind/pybind11

  Seamless operability between C++11 and Python

  Language:C++16k2502.1k2.1k

• state-spaces/mamba

  Mamba SSM architecture

  Language:Python13.6k995781.2k

• AkashSingh3031/The-Complete-FAANG-Preparation

  Dive into this repository, a comprehensive resource covering Data Structures, Algorithms, 450 DSA by Love Babbar, Striver DSA sheet, Apna College DSA Sheet, and FAANG Questions! 🚀 That's not all! We've got Technical Subjects like Operating Systems, DBMS, SQL, Computer Networks, and Object-Oriented Programming, all waiting for you.

  Language:Jupyter Notebook10.9k170422.4k

• NVIDIA/TensorRT-LLM

  TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that contain state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT-LLM also contains components to create Python and C++ runtimes that execute those TensorRT engines.

  Language:C++9k972.1k1k

• CarperAI/trlx

  A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

  Language:Python4.6k51292472

• alxndrTL/mamba.py

  A simple and efficient Mamba implementation in pure PyTorch and MLX.

  Language:Python1.1k84798

• twitter-archive/torch-autograd

  Autograd automatically differentiates native Torch code

  Language:Lua55950108115

• ccsb-scripps/AutoDock-GPU

  AutoDock for GPUs and other accelerators

  Language:C++41915210114

• proroklab/VectorizedMultiAgentSimulator

  VMAS is a vectorized differentiable simulator designed for efficient Multi-Agent Reinforcement Learning benchmarking. It is comprised of a vectorized 2D physics engine written in PyTorch and a set of challenging multi-robot scenarios. Additional scenarios can be implemented through a simple and modular interface.

  Language:Python35996072

• google/grain

  Library for reading and processing ML training data.

  Language:Python344122526

• shacklettbp/madrona

  Language:C++32991232

• pengxingang/Pocket2Mol

  Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pockets

  Language:Python27494072

• maabuu/posebusters

  Plausibility checks for generated molecule poses.

  Language:Python23332014

• justinchiu/openlogprobs

  Extract full next-token probabilities via language model APIs

  Language:Python2303114

• MichaelTMatthews/Craftax

  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.

  Language:Python22231422

• oneapi-src/oneCCL

  oneAPI Collective Communications Library (oneCCL)

  Language:C++214224372

• corl-team/xland-minigrid

  JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️

  Language:Python19491515

• soniajoseph/ViT-Prisma

  ViT Prisma is a mechanistic interpretability library for Vision Transformers (ViTs).

  Language:Jupyter Notebook19096119

• proger/accelerated-scan

  Accelerated First Order Parallel Associative Scan

  Language:Python166888

• instadeepai/og-marl

  Datasets with baselines for offline multi-agent reinforcement learning.

  Language:Python15571513

• toshas/torch-discounted-cumsum

  Fast Discounted Cumulative Sums in PyTorch

  Language:Python95288

• luchris429/popjaxrl

  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]

  Language:Python924712

• JoeyTeng/jaxrenderer

  Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrenderer/wiki; PR: https://github.com/google/brax/pull/367

  Language:Jupyter Notebook66457

• ZJLAB-AMMI/LLM4RL

  A RL approach to enable cost-effective, intelligent interactions between a local agent and a remote LLM

  Language:Python662413

• Gothos/LRU-pytorch

  Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch

  Language:Python63414

• NicolasZucchet/minimal-LRU

  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)

  Language:Python48156

• Cognitive-AI-Systems/learn-to-follow

  [AAAI-2024] Follower: This study addresses the challenging problem of decentralized lifelong multi-agent pathfinding. The proposed Follower approach utilizes a combination of a planning algorithm for constructing a long-term plan and reinforcement learning for resolving local conflicts.

  Language:C++37322

• johnryan465/pscan

  Language:Python36113

• Sanfee18/singularity-minerl

  A Singularity container that allows you to execute scripts on GPU inside headless environments.

  Language:Shell311