Scientific Ml Python Packages

jet-for-pytorch

Taylor-mode automatic differentiation for PyTorch

arxiv-embedding-benchmark

Benchmark embedding models on academic paper similarity and retrieval tasks.

pinn-rk

Runge–Kutta Physics-Informed Neural Networks with time-discrete losses (Gauss/Radau/Lobatto) in PyTorch.