Lightning fast C++/CUDA neural network framework
Sandbox for training deep learning networks
A Julia framework for invertible neural networks
Scalable and user friendly neural forecasting algorithms.
Message Passing Neural Networks for Molecule Property Prediction
Physics-Informed Neural Networks (PINN) Solvers
Internet-scale Neural Networks
BitNet: Scaling 1-bit Transformers for Large Language Models
High-performance neural network inference framework for mobile
DNN && GAN && NLP && BIG DATA
Pre-built implicit layer architectures with O(1) backprop, GPUs
Open source machine learning framework
Python-based neural networks API
SpikingJelly is an open-source deep learning framework
Simulation of spiking neural networks (SNNs) using PyTorch
Training neural networks on Apple Neural Engine via APIs
A JAX research toolkit to build, edit, & visualize neural networks
Deep learning driven jazz generation using Keras & Theano
DepGraph: Towards Any Structural Pruning
CV, NLP, LLM project applications, and advanced engineering deployment
Pre-trained Neural Network models in Axon
Global weather forecasting model using graph neural networks and JAX
Handwritten Text Recognition (HTR) system implemented with TensorFlow
Neural Network primitives with multiple backends