Open-source, high-performance AI model with advanced reasoning
Jittor is a high-performance deep learning framework
An Easy-to-use, Scalable and High-performance RLHF Framework
RL implementations
Powerful AI language model (MoE) optimized for efficiency/performance
SAPIEN Manipulation Skill Framework
C++-based high-performance parallel environment execution engine
A modular high-level library to train embodied AI agents
Advanced evolutionary computation library built on top of PyTorch
Massively parallel rigidbody physics simulation
A modular, primitive-first, python-first PyTorch library
VMAS is a vectorized differentiable simulator
PyTorch version of Stable Baselines
Algorithms for explaining machine learning models
LV10: High-coherence C++ lattice.
A collection of reference Jupyter notebooks and demo AI/ML application
High-quality single-file implementations of SOTA Offline
A high-performance distributed training framework
Applications of Deep Neural Networks
TradeMaster is an open-source platform for quantitative trading
Massively Parallel Deep Reinforcement Learning
High-quality single file implementation of Deep Reinforcement Learning
Deep learning and reinforcement learning library for scientists
Simple and comprehensive tutorials in TensorFlow