Zer0112's Stars
rasbt/LLMs-from-scratch
Implement a ChatGPT-like LLM in PyTorch from scratch, step by step
Azure/PyRIT
The Python Risk Identification Tool for generative AI (PyRIT) is an open source framework built to empower security professionals and engineers to proactively identify risks in generative AI systems.
ziglang/zig
General-purpose programming language and toolchain for maintaining robust, optimal, and reusable software.
pre-commit/pre-commit
A framework for managing and maintaining multi-language pre-commit hooks.
facebookresearch/xformers
Hackable and optimized Transformers building blocks, supporting a composable construction.
facebookincubator/AITemplate
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
labmlai/annotated_deep_learning_paper_implementations
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
facebookresearch/torchdim
Named tensors with first-class dimensions for PyTorch
mosaicml/composer
Supercharge Your Model Training
nebuly-ai/nos
Module to Automatically maximize the utilization of GPU resources in a Kubernetes cluster through real-time dynamic partitioning and elastic quotas - Effortless optimization at its finest!
hpcaitech/ColossalAI
Making large AI models cheaper, faster and more accessible
pytorch/rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
michaelnny/deep_rl_zoo
A collection of Deep Reinforcement Learning algorithms implemented with PyTorch to solve Atari games and classic control tasks like CartPole, LunarLander, and MountainCar.
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
thomyphan/darts-coding
DARTS Coding Sessions
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
jwasham/coding-interview-university
A complete computer science study plan to become a software engineer.
JuliaReinforcementLearning/ReinforcementLearning.jl
A reinforcement learning package for Julia
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
HigherOrderCO/HVM
A massively parallel, optimal functional runtime in Rust
sympy/sympy
A computer algebra system written in pure Python
einsteinpy/einsteinpy
Repository for the EinsteinPy core package :rocket:
pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
tensorflow/tensorflow
An Open Source Machine Learning Framework for Everyone