YifanShenSZ/coreai-models
Model export recipes, Python primitives, and Swift runtime utilities for on-device AI
Core AI | ex PyTorch
Repositories
YifanShenSZ/Fortran-Library
Mathematical & chemical routines e.g. nonlinear optimization etc. with c++ & python frontend
YifanShenSZ/coreai-torch
Bridges PyTorch and Core AI. Convert existing models to Core AI IR, or author new ones from PyTorch via composite ops, custom op lowerings, and inline Metal GPU kernels.
YifanShenSZ/coreai-optimization
A library for PyTorch model compression and optimizations for deployment via Core AI on Apple silicon.
YifanShenSZ/executorch
End-to-end solution for enabling on-device AI across mobile and edge devices for PyTorch models
YifanShenSZ/coremltools
Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.
YifanShenSZ/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
YifanShenSZ/ao
PyTorch native quantization and sparsity for training and inference
YifanShenSZ/transformers
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
YifanShenSZ/YifanShenSZ
YifanShenSZ/PyTorch-Benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
YifanShenSZ/tensorflow
An Open Source Machine Learning Framework for Everyone
YifanShenSZ/vibronics
Vibronic spectrum simulation package
YifanShenSZ/diabatz
Global diabatization of ab initio wave functions by machine learning schemes
YifanShenSZ/C6H7N-aniline
Project on aniline photodissociation
YifanShenSZ/SurfGenBound
Surface generation package for bound system
YifanShenSZ/Cpp-Library
Command line argument parser & basic templates & operator overloads etc.
YifanShenSZ/YifanShenSZ.github.io
YifanShenSZ/pytorch-notes
some random notes I took during learning pytorch
YifanShenSZ/tutorials
PyTorch tutorials.
YifanShenSZ/Torch-Chemistry
Chemistry kernel library based on LibTorch
YifanShenSZ/MC
Monte Carlo simulation package for canonical ensembles
YifanShenSZ/OneDimDVR
Numerically solve the time dependent schrodinger equation by discrete variable representation method
YifanShenSZ/Foptim
Fortran nonlinear optimization library with C++ frontend
YifanShenSZ/hop
Surface hopping with diabatic potential energy matrix
YifanShenSZ/Tool-Collection
Collection of useful tools in my research like data visualization etc.
YifanShenSZ/Columbus7-Extension
Useful extension and modification for Columbus7
YifanShenSZ/MCSMD
Multi-centred semiclassical Moyal dynamics
YifanShenSZ/critics
A critical geometry searcher
YifanShenSZ/SurfGenTorch
Surface generation package based on libtorch