This job has closed.

BayOne Solutions · 9 hours ago

Machine Learning Engineer(LLM/Python)

San Jose, CA

Contract

Hybrid

Mid Level

$60/hr - $70/hr

BayOne Solutions is seeking a hands-on Machine Learning Engineer with a strong computer science foundation and practical experience building LLMs. The role involves designing, training, optimizing, and deploying LLMs for real-world applications, as well as collaborating with cross-functional teams to apply models to various use cases.

Information and Communications Technology (ICT)Information ServicesInformation TechnologySaaSSoftware

Growth Opportunities

H1B Sponsor Likely

Hiring Manager

Neeraj Maurya

Responsibilities

Design, train, and fine-tune specific LLMs for enterprise use cases

Build robust data pipelines for large-scale text preprocessing and model training

Optimize LLMs for efficiency using techniques such as LoRA, quantization, and distributed training

Deploy and serve models in production environments using modern ML infrastructure

Collaborate with the AI Innovation core team to research, test, and scale new LLM architectures and techniques

Work closely with cross-functional teams to apply models to security assessment and future CX use cases

Qualification

PythonLLM experiencePyTorchTransformer architecturesDistributed training frameworksModel deployment toolsMLOps workflowsVector databasesCloud ML platformsSecurity/compliance-aware MLCollaboration skillsCommunication skills

Required

Bachelor's or Master's degree in Computer Science, Electrical Engineering, or related field

Strong programming skills in Python (with C++/Java as a plus)

Hands-on experience building and training LLMs (small, medium, or large scale)

Deep understanding of transformer architectures, tokenization, embeddings, and pretraining vs. fine-tuning methods

Proficiency with PyTorch (primary) and familiarity with Hugging Face ecosystem

Experience with distributed training frameworks (DeepSpeed, FSDP, DDP, Accelerate, Horovod)

Knowledge of model deployment tools (ONNX, TorchScript, Triton Inference Server, FastAPI)

Preferred

Familiarity with MLOps workflows (MLflow, Weights & Biases, ClearML)

Knowledge of vector databases (FAISS, Pinecone, Weaviate, Milvus) and RAG pipelines

Experience with cloud ML platforms (AWS SageMaker, Azure ML, GCP Vertex AI)

Background in security/compliance-aware ML (prompt injection defense, data leakage prevention)

Collaboration and communication skills to work with local and global teams

Company

BayOne Solutions

BayOne Solutions provides computer programming services.

Founded in 2012

Pleasanton, California, USA

501-1000 employees

https://bayone.com/

H1B Sponsorship

BayOne Solutions has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (23)

2024 (25)

2023 (20)

2022 (30)

2021 (20)

2020 (37)