Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for pytorch jax
jax
x
pytorch
x
77 search results found
Transformers
⭐
124,049
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Keras
⭐
60,949
Deep Learning for humans
D2l En
⭐
20,613
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
Best Of Ml Python
⭐
14,990
🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
Ivy
⭐
13,904
The Unified AI Framework
Wandb
⭐
8,204
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
Einops
⭐
7,548
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Pyprobml
⭐
6,066
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Interviews.ai
⭐
3,146
It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; however, it is my hope that even the most experienced researchers will find it fascinating as well.
Thinc
⭐
2,774
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
Foolbox
⭐
2,600
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
Iree
⭐
2,303
A retargetable MLIR-based machine learning compiler and runtime toolkit.
Deepxde
⭐
2,135
A library for scientific machine learning and physics-informed learning
Pennylane
⭐
2,117
PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.
Uvadlc_notebooks
⭐
1,860
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022
Tensorly
⭐
1,466
TensorLy: Tensor Learning in Python.
Training Operator
⭐
1,447
Distributed ML Training and Fine-Tuning on Kubernetes
Big_transfer
⭐
1,421
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
Transformerengine
⭐
1,172
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
Mosec
⭐
661
A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine
Eagerpy
⭐
645
PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
Tensor Sensor
⭐
634
The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
Treasure Of Transformers
⭐
541
💁 Awesome Treasure of Transformers Models for Natural Language processing contains papers, videos, blogs, official repo along with colab Notebooks. 🛫☑️
Pdebench
⭐
522
PDEBench: An Extensive Benchmark for Scientific Machine Learning
Ml Aim
⭐
515
This repository provides the code and model checkpoints of the research paper: Scalable Pre-training of Large Autoregressive Image Models
Fast Soft Sort
⭐
476
Fast Differentiable Sorting and Ranking
Elegy
⭐
430
A High Level API for Deep Learning in JAX
Neuralcompression
⭐
430
A collection of tools for neural compression enthusiasts.
Dataloader
⭐
390
The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX
Machine Learning Guide
⭐
366
Machine learning Guide. Learn all about Machine Learning Tools, Libraries, Frameworks, Large Language Models (LLMs), and Training Models.
Tevatron
⭐
334
Tevatron - A flexible toolkit for neural retrieval research and development.
Learning Python Physics Informed Machine Learning Pinns Deeponets
⭐
297
Physics Informed Machine Learning Tutorials (Pytorch and Jax)
Pyhpc Benchmarks
⭐
290
A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python 🚀
Scimlbenchmarks.jl
⭐
279
Scientific machine learning (SciML) benchmarks, AI for science, and (differential) equation solvers. Covers Julia, Python (PyTorch, Jax), MATLAB, R
Squirrel Core
⭐
271
A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way 🌰
Pyhf
⭐
265
pure-Python HistFactory implementation with tensors and autodiff
Skrl
⭐
263
Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Isaac Orbit and Omniverse Isaac Gym
Gradcache
⭐
254
Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
Tensorcircuit
⭐
230
Tensor network based quantum software framework for the NISQ era
Funsor
⭐
224
Functional tensors for probabilistic programming
Sunstreaker
⭐
126
以jax为后端的类似keras的框架
Efficientnet Jax
⭐
121
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
Efficient Vdvae
⭐
115
Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"
Swanlab
⭐
108
🧐SwanLab: track and visualize all the pieces of your machine learning pipeline. Join our WeChat ⬇️
Autoray
⭐
106
Abstract your array operations.
Bayesde
⭐
101
Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
Robustness Vit
⭐
97
Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).
Easydel
⭐
83
EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥
Adam
⭐
82
adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.
Dm_nevis
⭐
71
NEVIS'22: Benchmarking the next generation of never-ending learners
Vision
⭐
69
3D Vision functions with end-to-end support for deep learning developers, written in Ivy.
Hyper Nn
⭐
65
Easy Hypernetworks in Pytorch and Jax
Gym
⭐
55
Fully differentiable RL environments, written in Ivy.
Dks
⭐
50
Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to train.
Madam
⭐
45
Pytorch and Jax code for the Madam optimiser.
Rockpool
⭐
37
A machine learning library for spiking neural networks. Supports training with both torch and jax pipelines, and deployment to neuromorphic hardware.
Meent
⭐
34
Electromagnetic simulation (RCWA) & optimization package in Python
Pycallchainrules.jl
⭐
34
Differentiate python calls from Julia
Tensorshare
⭐
25
🤝 Trade any tensors over the network
Jax Dataloader
⭐
20
Pytorch-like dataloaders in JAX.
Nixos Nvidia Cuda Python Docker Compose
⭐
18
A step-by-step guide to setting up Nvidia GPUs with CUDA support running on Docker (and Compose) containers on NixOS host
Mech
⭐
14
Mechanics functions with end-to-end support for deep learning developers, written in Ivy.
Scimlbenchmarksoutput
⭐
13
SciML-Bench Benchmarks for Scientific Machine Learning (SciML), Physics-Informed Machine Learning (PIML), and Scientific AI Performance
Deep Learning Projects
⭐
13
Best Deep Learning Projects for Advanced Learners
Whisper Finetune
⭐
13
Fine-tune and evaluate Whisper models for Automatic Speech Recognition (ASR) on custom datasets or datasets from huggingface.
Dataset_grouper
⭐
12
Sphericalharmonics
⭐
12
Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax
Detect Fake Text
⭐
12
LLM - Detect AI Generated Text || Identify which essay was written by a large language model
Nvidia Gpu Ml Library Test
⭐
11
Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being properly picked up
Yerbamate
⭐
10
A framework-agnostic deep learning package and experiment manager
Nablafuzz
⭐
9
Fuzzing Automatic Differentiation in Deep-Learning Libraries (ICSE'23)
Holographic Reduced Representations
⭐
9
Holographic Reduced Representations
Rl Basics
⭐
8
Simple single file implementation of Deep Reinforcement Learning algorithms
Animus
⭐
7
Minimalistic framework to run machine learning experiments.
Doc Builder
⭐
6
Simple Documentation Builder for Ivy Projects.
Differentiable_t1d
⭐
5
Differentiable simulators for Type 1 Diabetes research in Jax and PyTorch
Autodiff101
⭐
5
An introduction to Automatic Differentiation with theory and code examples.
Related Searches
Python Pytorch (15,943)
Deep Learning Pytorch (7,533)
Jupyter Notebook Pytorch (4,892)
Machine Learning Pytorch (2,934)
Dataset Pytorch (1,848)
Pytorch Convolutional Neural Networks (1,777)
Pytorch Neural Network (1,631)
Pytorch Natural Language Processing (1,408)
Pytorch Computer Vision (1,230)
Pytorch Neural (1,217)
1-77 of 77 search results
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2024 Awesome Open Source. All rights reserved.