Rust bindings for the C++ api of PyTorch.

Machine learning Calabi-Yau metrics with JAX

A python library for studying Calabi-Yau metrics

Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures

A RWKV management and startup tool, full automation, only 8MB. And provides an interface compatible with the OpenAI API. RWKV is a large language model that is fully open source and available for c…

RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference,…

INT4/INT5/INT8 and FP16 inference on CPU for RWKV language model

Rust bindings for the Python interpreter

ndarray: an N-dimensional array with array views, multidimensional slicing, and efficient operations

Understanding the paper "Principles of Riemannian Geometry in Neural Networks" by Michael Hauser and Asok Ray

Minimal code example of the publication "On the accuracy, stability and computational efficiency of explicit last-stage diagonally implicit Runge-Kutta methods (ELDIRK) for the adaptive solution of…

Neuromorphic Intermediate Representation reference implementation

A deep learning library for spiking neural networks which is based on PyTorch, focuses on fast training and supports inference on neuromorphic hardware.

Implementation of 'Inverse-design of nonlinear mechanical metamaterials via video denoising diffusion models'

Code for 'Physics-Informed Neural Networks for Shell Structures'

Deep learning with spiking neural networks (SNNs) in PyTorch.

Deep and online learning with spiking neural networks in Python

Computations and statistics on manifolds with geometric structures.

Fast and Easy Infinite Neural Networks in Python

Physics-Informed Neural Networks (PINN) Solvers of (Partial) Differential Equations for Scientific Machine Learning (SciML) accelerated simulation